Update app.py

app.py

@@ -8,14 +8,14 @@ from datetime import datetime
 
 # Model which is used
 checkpoint = "CohereForAI/c4ai-command-r-v01"
-checkpoint = "mistralai/Mistral-7B-Instruct-v0.2"
-checkpoint = "google/flan-t5-small"
+checkpoint = "mistralai/Mistral-7B-Instruct-v0.1"
+#checkpoint = "google/flan-t5-small"
 path_to_log = "FlaggedFalse.txt"
 
 # Inference client with the model (And HF-token if needed)
 client = InferenceClient(checkpoint)
 tokenizer = AutoTokenizer.from_pretrained(checkpoint)
-if checkpoint == "mistralai/Mistral-7B-Instruct-v0.2":
+if checkpoint == "mistralai/Mistral-7B-Instruct-v0.1":
     # Tokenizer chat template correction(Only works for mistral models)
     chat_template = open("mistral-instruct.jinja").read()
     chat_template = chat_template.replace('    ', '').replace('\n', '')