sync

app.py

@@ -67,7 +67,7 @@ with gr.Blocks(theme='freddyaboulton/test-blue') as demo:
         gr.Markdown("<center><h3>AI Response</h3></center>")
         image_output_box = gr.Image(type="pil", label="Final Generated Image")
 
-    submit_btn.click(fn=generate_image, inputs = [prompt,neg_prompt,source_image], outputs=[image_output])
+    submit_btn.click(fn=generate_image, inputs = [prompt,neg_prompt,source_image], outputs=[image_output_box])
     clear_btn.click(fn=clear_fields,outputs=[prompt,neg_prompt,image_output_box])
 
 demo.launch()