Update app.py

app.py

@@ -43,9 +43,10 @@ def predict(dict, prompt="high quality, best quality", negative_prompt="", guida
     generator=generator,
     height=int(height), width=int(width),
     guidance_scale=guidance_scale
-    ).images
+    )
 
-    return output[0] #, gr.update(visible=True)
+    torch.cuda.empty_cache
+    return output.images[0] #, gr.update(visible=True)
 
 
 css = '''
@@ -98,7 +99,8 @@ with image_blocks as demo:
         ''')
     with gr.Row():
                 with gr.Column():
-                    image = gr.Image(sources=['upload'], elem_id="image_upload", type="pil", label="Upload", height=400)
+                    image = gr.Image(sources=None, 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")