Update app.py

app.py

@@ -77,7 +77,7 @@ def generate_caption_and_image(image):
     # Generate image based on the caption
     generated_image = pipe(prompt).images[0]
 
-    return caption, generated_image
+    return prompt, generated_image
 
 # Gradio UI
 iface = gr.Interface(