Update app.py

app.py

@@ -4,7 +4,6 @@ f = modal.Cls.lookup("casa-interior-hf-v1", "DesignModel")
 
 def casa_ai_run(dict=None, text=None):
 
-    print(dict.keys())
     image = dict["background"].convert("RGB")
     mask = dict["layers"][0].convert("RGB")
     
@@ -21,7 +20,7 @@ title = "Casa-AI Demo"
 description = "A Gradio interface to use CasaAI for virtual staging"
 
 demo = gr.Interface(casa_ai_run,
-                    inputs = [gr.ImageEditor(sources='upload', brush=gr.Brush(colors=["#000000"]), elem_id="image_upload", type="pil", label="Upload", layers=False, eraser=False),
+                    inputs = [gr.ImageEditor(sources='upload', brush=gr.Brush(colors=["#000000"]), elem_id="image_upload", type="pil", label="Upload", layers=False, eraser=True, transforms=[]),
                               gr.Textbox(label="Prompt to design room"), 
                               ],
                     outputs = [