Update app.py

app.py

@@ -132,10 +132,9 @@ def generate(
         yield output
 
     final_story = "".join(outputs)
-    image_url = generate_image(final_story)
-    return f"{final_story}\n\n![Generated Image]({image_url})"
+    markdown_image = f"![Generated Image]({image_url})"
+    return f"{response_text}\n\n{markdown_image}"
 
-![Description](URL)
 
 chat_interface = gr.ChatInterface(
     fn=generate,