Update app.py

app.py

@@ -320,7 +320,7 @@ with gr.Blocks() as iface:
         with gr.Column():
             sample_choice = gr.Radio(choices=["Sample 1", "Sample 2"], label="Choose a Sample Background")
             input_image = gr.Image(type="pil", label="Input Image")
-            mask_input = gr.ImageEditor(type="pil", label="Draw Mask on Image", interactive=True)
+            mask_input = gr.Image(type="pil", label="Draw Mask on Image", tool="sketch", interactive=True)
             text_input = gr.Textbox(label="Text to Synthesize / Edit")
             outlined_image = gr.Image(type="pil", label="Original Image with Mask Outline")
             
@@ -355,10 +355,10 @@ with gr.Blocks() as iface:
         inputs=[sample_choice],
         outputs=[input_image]
     )
-
+    
     # Update mask when input image changes
     input_image.change(
-        lambda image: None if image is None else {"image": image},
+        lambda image: image,  # Pass through image to mask_input
         inputs=[input_image],
         outputs=[mask_input]
     )