tt

app.py

@@ -4,8 +4,8 @@ import gradio as gr
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-model_name_a = "meta-llama/Llama-2-7b-chat-hf"
-model_name_b = "meta-llama/Llama-2-7b-chat-hf"  # you can replace this with a second different model or finetuned variant
+model_name_a = "mistralai/Mistral-7B-v0.1"
+model_name_b = "mistralai/Mistral-7B-v0.1"  # you can replace this with a second different model or finetuned variant
 
 print("Loading tokenizer...")
 tokenizer = AutoTokenizer.from_pretrained(model_name_a)