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 == "M4-ai/tau-0.5B":
+    if model_name == "M4-ai/tau-1.8B":
         prompt = user_input
     else:
         prompt = f"<|im_start|>user\n{user_input}<|im_end|>\n<|im_start|>assistant\n"