Update app.py

app.py

@@ -170,9 +170,9 @@ with gr.Blocks(css="footer{display:none !important}") as demo:
                     history.append([message, full_response])
 
                     # Generate image based on the latest assistant response
-                    image = generate_image(text_response)
+                    #image = generate_image(text_response)
                     
-                    return history, "",image
+                    return history, ""
 
                 user_input.submit(
                     wrapped_predict,