Update app.py

app.py

@@ -42,21 +42,23 @@ chat = utils.ChatState(gemma_lm) # adding a chat helper to manage the conversati
 def launch(message):
 
     # Uncomment for QA system without chat history/memory
-    #template = "Instruction:\n{instruction}\n\nResponse:\n{response}"
-    #prompt = template.format(
-    #    instruction=message,                                                                   
-    #    response="",
-    #)
-    #out = gemma_lm.generate(prompt, max_length=1024)
-    #ind = out.index('\n\nResponse:\n') + len('\n\nResponse:\n')
-    #return out[ind:]
+    #
+    template = "Instruction:\n{instruction}\n\nResponse:\n{response}"
+    prompt = template.format(
+        instruction=message,                                                                   
+        response="",
+    )
+    out = gemma_lm.generate(prompt, max_length=1024)
+    ind = out.index('\n\nResponse:\n') + len('\n\nResponse:\n')
+    return out[ind:]
     
     # This is to convert QA system to a chatbot
-    return chat.send_message(message) + "\n\n##### History #####\n\n" +  chat.get_history() 
+    #chat_history = chat.send_message(message) + "\n\n##### History #####\n\n" +  chat.get_history()  
+    #return chat_history
 
 iface = gr.Interface(launch,
-                     inputs=["textbox",gr.State(value=[])],
-                     outputs=["textbox",gr.State(value=[])],
+                     inputs="text",
+                     outputs="text",
                      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.")