Update app.py (#2)

- Update app.py (a8e839c595dc107ae3c2d898ec69422e5465ae3f)


Co-authored-by: Russel Brunton <[email protected]>

app.py

@@ -352,6 +352,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: