Update app.py

app.py

@@ -52,11 +52,11 @@ def launch(message):
     #return out[ind:]
     
     # This is to convert QA system to a chatbot
-    return chat.send_message(message) + "\nhistory\n" +  chat.get_history() 
+    return chat.send_message(message) + "\n\n##### History #####\n\n" +  chat.get_history() 
 
 iface = gr.Interface(launch,
-                     inputs="text",
-                     outputs="text",
+                     inputs=["text",gr.State()],
+                     outputs=["text",gr.State()],
                      css=css,
                      title="Hey I am CosmoGemma 👋 I can answer cosmology questions from astroph.CO research articles. Try me :)",
                      description="Gemma_2b_en fine-tuned on QA pairs (~3.5k) generated from Cosmology and Nongalactic Astrophysics articles (arXiv astro-ph.CO) from 2018-2022 and tested on QA pairs (~1k) generated from 2023 articles, scoring over 75% accuracy.")