Update app.py

app.py

@@ -26,7 +26,7 @@ def generate(
         tokenizer.eos_token = "<|im_end|>"
         print(tokenizer)
         pipe.tokenizer = tokenizer
-        if model_name == "Locutusque/TinyMistral-248M":
+        if model_name == "Locutusque/TinyMistral-248M-v3":
             prompt = " ".join(history) + message
         else:
             prompt = f"<|im_start|>system\n{system}<|im_end|>\n"
@@ -62,7 +62,7 @@ def generate(
         print(f"An error occurred: {e}")
         yield "An error occurred during generation."
 
-model_choices = ["Locutusque/Apollo-2.0-Llama-3.1-8B", "Locutusque/TinyMistral-248M", "Locutusque/Hercules-6.1-Llama-3.1-8B", "Locutusque/DareQwen-2.5-7B", "Locutusque/Hercules-5.0-Index-1.9B", "Locutusque/StockQwen-2.5-7B"]
+model_choices = ["Locutusque/Apollo-2.0-Llama-3.1-8B", "Locutusque/TinyMistral-248M-v3", "Locutusque/Hercules-6.1-Llama-3.1-8B", "Locutusque/DareQwen-2.5-7B", "Locutusque/Hercules-5.0-Index-1.9B", "Locutusque/StockQwen-2.5-7B"]
 # What at the best options? 
 g = gr.ChatInterface(
     fn=generate,