Update app.py

app.py

@@ -133,10 +133,7 @@ def generate(
 
     final_story = "".join(outputs)
     image_url = generate_image(final_story)
-    yield image_url
-
-
-
+    outputs.append(image_url)
 
 chat_interface = gr.ChatInterface(
     fn=generate,
@@ -147,7 +144,7 @@ chat_interface = gr.ChatInterface(
         ["Could you please provide an explanation about the concept of recursion?"],
         ["Could you explain what a URL is?"]
     ],
-    theme='shivi/calm_seafoam',autofocus=True
+    theme='shivi/calm_seafoam',autofocus=True, 
 )
 
 # Gradio Web Interface