Limit upload image size

app.py

@@ -15,6 +15,7 @@ from super_resolve import process
 
 REPO_ID_EDSR = "prs-eth/thera-edsr-pro"
 REPO_ID_RDN = "prs-eth/thera-rdn-pro"
+MAX_SIZE = 600
 
 print(f"JAX devices: {jax.devices()}")
 print(f"JAX device type: {jax.devices()[0].device_kind}")
@@ -94,6 +95,10 @@ class TheraApp(DualVisionApp):
         do_ensemble = kwargs.get("do_ensemble", self.DEFAULT_DO_ENSEMBLE)
         model = kwargs.get("model", self.DEFAULT_MODEL)
 
+        if max(*image_in.size) > MAX_SIZE:
+            raise gr.Error(f"We have currently limited the size of uploaded images to {MAX_SIZE}x{MAX_SIZE}"
+                f" pixels, to enable a smooth experience to all users.")
+
         source = np.asarray(image_in) / 255.
 
         # determine target shape