Update app.py

crop & resize input image on .upload

app.py

@@ -192,7 +192,16 @@ def randomize_seed_fn(seed, randomize_seed):
     torch.manual_seed(seed)
     return seed
 
-    
+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
     
 
 def get_example():
@@ -716,7 +725,7 @@ with gr.Blocks(css="style.css") as demo:
         outputs = [do_inversion],
         queue = False)
     # Automatically start inverting upon input_image change
-    input_image.upload(
+    input_image.upload(fn = crop_image, inputs = [input_image], outputs = [input_image],queue=False).then(
         fn = reset_do_inversion,
         outputs = [do_inversion],
         queue = False).then(fn = caption_image,