Update app.py

app.py

@@ -19,7 +19,7 @@ def generate(
     pipe = load_model(model_name)
 
     # Set tokenize correctly. Otherwise ticking the box breaks it.
-    if model_name == "Locutusque/TinyMistral-248M-v3":
+    if model_name == "M4-ai/tau-0.5B":
         prompt = user_input
     else:
         prompt = f"<|im_start|>user\n{user_input}<|im_end|>\n<|im_start|>assistant\n"
@@ -27,7 +27,7 @@ def generate(
                    temperature=temperature, top_k=top_k, top_p=top_p, repetition_penalty=1.10)
     return outputs[0]["generated_text"]
 
-model_choices = ["M4-ai/hyperion-medium-preview", "M4-ai/NeuralReyna-Mini-1.8B-v0.2", "aloobun/Cypher-Laser-Mixtral-2x1.8B-v0.1", "Locutusque/NeuralHyperion-Medium-Preview", "aloobun/Cypher-Mini-1.8B", "Locutusque/Hercules-2.0-Qwen1.5-1.8B", "Locutusque/Hercules-2.5-Mistral-7B", "Locutusque/hyperion-small-preview"]
+model_choices = ["M4-ai/hyperion-medium-preview", "M4-ai/NeuralReyna-Mini-1.8B-v0.2", "aloobun/Cypher-Laser-Mixtral-2x1.8B-v0.1", "Locutusque/NeuralHyperion-Medium-Preview", "aloobun/Cypher-Mini-1.8B", "Locutusque/Hercules-2.0-Qwen1.5-1.8B", "Locutusque/Hercules-2.5-Mistral-7B", "M4-ai/tau-0.5B"]
 # What at the best options? 
 g = gr.Interface(
     fn=generate,