Update app.py

app.py

@@ -30,7 +30,8 @@ def predict(image, prompt="high quality, best quality", negative_prompt="", guid
     if negative_prompt == "":
         negative_prompt = None
     
-    init_image = image.convert("RGB")
+    init_image = image['image'].convert("RGB")    
+    # init_image = image.convert("RGB")
     # if center_crop is False:
     # init_image = init_image.resize((224, 224))
     
@@ -100,7 +101,7 @@ with image_blocks as demo:
         ''')
     with gr.Row():
                 with gr.Column():
-                    image = gr.ImageEditor(elem_id="image_upload", type="pil", label="Upload", height=400, crop_size="1:1")
+                    image = gr.ImageEditor(elem_id="image_upload", type="pil", label="Upload")
                     # image = gr.Image(sources=None, type="pil") # None for upload, ctrl+v and webcam