Update app.py

app.py

@@ -4,8 +4,9 @@ import torch
 from transformers import AutoTokenizer, AutoModel
 import plotly.graph_objects as go
 
-# Update the model name to Llama 3.1
-model_name = "meta-llama/Meta-Llama-3.1-405B-Instruct-FP8"
+TOKEN = os.getenv("HF_TOKEN")
+
+model_name = "mistralai/Mistral-Large-Instruct-2407"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = None