Update run.py

run.py

@@ -16,7 +16,7 @@ def response(message, history, model):
   params={"max_length":500, "return_full_text":False} #, "stream":True
   url = f"https://api-inference.huggingface.co/models/{model_id}"
   correction=1
-  prompt=f"[INST] {message} [/INST]" 
+  prompt=f"<s>[INST] {message} [/INST]" 
   print("URL: "+url)
   print(params)
   print("User: "+message+"\nAI: ")
@@ -42,6 +42,8 @@ x=requests.get(f"https://api-inference.huggingface.co/framework/text-generation-
 x=[i["model_id"] for i in x.json()]
 x.insert(0,"Default")
 print(x)
+x=[s for s in x if s.startswith("mistral")]
+print(x)
 
 gr.ChatInterface(
   response,