Update app.py

app.py

@@ -97,25 +97,16 @@ prompt_template = PromptTemplate.from_template('''system role :{context} \
         assistance: 
         ''')
 
-import random
-import time
-
 # Define Gradio Interface
-with gr.Blocks() as chat:
-    chatbot = gr.Chatbot()
-    msg = gr.Textbox()
-    clear = gr.ClearButton([msg, chatbot])
-
-    def respond(message, chat_history):
-        # Use the global conversation variable here
-        response = conversation.run(message)
-        chat_history.append((message, response))
-        return response, chat_history
-
-    msg.submit(respond, [msg, chatbot], [msg, chatbot])
+iface = gr.Interface(
+    fn=lambda query: conversation.run(prompt_template.format(context=context, query=query)),
+    inputs=gr.Textbox(),
+    outputs=gr.Textbox(),
+    live=True,
+    capture_session=True
+)
 
 # Launch Gradio Interface
-if __name__ == "__main__":
-    chat.launch()
+iface.launch()
 
 # gr.load("models/ksh-nyp/llama-2-7b-chat-TCMKB").launch()