Update app.py

app.py

@@ -27,7 +27,7 @@ mistral_models=["mistralai/Mixtral-8x7B-Instruct-v0.1"]
 # Inference client with the model (And HF-token if needed)
 client = InferenceClient(checkpoint)
 tokenizer = AutoTokenizer.from_pretrained(checkpoint)
-if checkpoint == "NousResearch/Nous-He-8x7B-DPO":
+if checkpoint in mistral_models:
     # Tokenizer chat template correction(Only works for mistral models)
     chat_template = open("mistral-instruct.jinja").read()
     chat_template = chat_template.replace('    ', '').replace('\n', '')