Update app.py

app.py

@@ -60,7 +60,7 @@ def fill_image(image, model_selection):
 
     margin = 100
     # Open the original image
-    source = image["image"]  # Changed from image["background"] to match new input format
+    source = image  # Changed from image["background"] to match new input format
     
     # Calculate new output size
     output_size = (source.width + 2*margin, source.height + 2*margin)
@@ -119,12 +119,9 @@ with gr.Blocks(css=css) as demo:
     run_button = gr.Button("Generate")
 
     with gr.Row():
-        input_image = gr.ImageMask(
+        input_image = gr.Image(
             type="pil",
             label="Input Image",
-            crop_size=(1024, 1024),
-            canvas_size=(1024, 1024),
-            layers=False,
             sources=["upload"],
         )