Update app.py

app.py

@@ -386,7 +386,16 @@ def file_upload(file):
     return image
     
     
-
+def crop_image(image):
+    h, w, c = image.shape
+    if h < w:
+        offset = (w - h) // 2
+        image = image[:, offset:offset + h]
+    elif w < h:
+        offset = (h - w) // 2
+        image = image[offset:offset + w]
+    image = np.array(Image.fromarray(image).resize((512, 512)))
+    return image
     
 intro = """
 <div style="display: flex;align-items: center;justify-content: center">
@@ -476,6 +485,13 @@ with gr.Blocks(css="style.css") as demo:
         
 
 
+        input_image.upload(
+        fn = crop_image,
+        inputs = [input_image],
+        outputs = [input_image],
+        queue=False,
+        concurrency_limit=None,
+        )
 
 
         invert_button.click(fn=run_inversion,