Update app.py

app.py

@@ -45,7 +45,7 @@ def predict(dict, prompt="high quality, best quality", negative_prompt="", guida
     guidance_scale=guidance_scale
     ).images
 
-    return output.images[0] #, gr.update(visible=True)
+    return output[0] #, gr.update(visible=True)
 
 
 css = '''
@@ -98,7 +98,7 @@ with image_blocks as demo:
         ''')
     with gr.Row():
                 with gr.Column():
-                    image = gr.Image(sources=['upload'], tool='sketch', elem_id="image_upload", type="pil", label="Upload", height=400)
+                    image = gr.Image(sources=['upload'], elem_id="image_upload", type="pil", label="Upload", height=400)
                     with gr.Row(elem_id="prompt-container", equal_height=True):
                         with gr.Row():
                             prompt = gr.Textbox(placeholder="Your prompt (you can leave it empty if you only want the image prompt as input)", show_label=False, elem_id="prompt")