Update app.py

app.py

@@ -1,14 +1,15 @@
 import gradio as gr
 from gradio_client import Client
 
-# Initialize the Gradio client
 client = Client("https://tonic1-mistralmed-chat.hf.space/--replicas/srl2h/")
 
+user_input_with_token = user_input + " </s>"
+
 def predict(user_input):
-    system_prompt = "You are MistralMed Created By Tonic-AI. You are a Very Helpful Assistant. You always provide a cautious answer, check your work, provide a long detailed answer."
+    system_prompt = "[INST]You are MistralMed Created By Tonic-AI. You are a Very Helpful Assistant. You always provide a cautious answer, check your work, provide a long detailed answer.[/INST]"
 
     result = client.predict(
-        user_input, 
+        user_input_with_token, 
         system_prompt,
         api_name="/predict"
     )