Update app_gradio.py

app_gradio.py

@@ -166,22 +166,7 @@ def generate_output(image, apply_filter, prompt: str, num_seeds: int = 3, lambda
     # Save the input image temporarily
     temp_image_path = os.path.join(exp_dir, "temp_input.png")
 
-    # Get original dimensions
-    width, height = image.size
-
-    # Step 1: Make the image square if it's not already
-    if width != height:
-        # Use the smaller dimension to crop to a square
-        min_dim = min(width, height)
-        left = (width - min_dim) // 2
-        top = (height - min_dim) // 2
-        right = left + min_dim
-        bottom = top + min_dim
-        image = image.crop((left, top, right, bottom))
-
-    # Step 2: Resize to 256 if the image is larger
-    if image.size[0] > 256:  # Image is square at this point
-        image = image.resize((256, 256), Image.LANCZOS)
+    image = image.resize((512, 512), Image.LANCZOS)
 
     image.save(temp_image_path)