Update app.py

app.py

@@ -350,6 +350,7 @@ class Head_Agent:
                 elif classify_query == python_code_num:
                     response = self.query_agent.process_query_response(self, user_query, 'python')
                     image, caption = self.answering_agent.generate_image(response)
+                    image = None
                     topic = "python"
                     st.session_state.stage = 0
                 else: