Crop and warn instead of error

app.py

@@ -15,7 +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
+MAX_SIZE = 500
 
 print(f"JAX devices: {jax.devices()}")
 print(f"JAX device type: {jax.devices()[0].device_kind}")
@@ -56,6 +56,9 @@ class TheraApp(DualVisionApp):
             <p align="center" style="margin-top: 0px;">
                 <strong>Upload a photo or select an example below to do arbitrary-scale super-resolution in real time!</strong>
             </p>
+            <p align="center" style="margin-top: 0px;">
+                <strong>Note: The model has not been trained on input images with JPEG artifacts, so this may not work well.</strong>
+            </p>
         """
         )
 
@@ -96,8 +99,15 @@ class TheraApp(DualVisionApp):
         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.")
+            gr.Warning(f"The image has been cropped to enable a smooth experience for all users.")
+            width, height = image_in.size
+            crop_width = min(width, MAX_SIZE)
+            crop_height = min(height, MAX_SIZE)
+            left = (width - crop_width) / 2
+            top = (height - crop_height) / 2
+            right = left + crop_width
+            bottom = top + crop_height
+            image_in = image_in.crop((left, top, right, bottom))
 
         source = np.asarray(image_in) / 255.