Update app.py

app.py

@@ -5,9 +5,11 @@ import torch
 
 # Load model and tokenizer
 def load_model():
-    model_name = "viet-ai/vistral-7b-chat"  # Vistral của Viet-Mistral
-    tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=os.getenv("HF_AUTH_TOKEN"))
-    model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=os.getenv("HF_AUTH_TOKEN"))
+    model_name = "Viet-Mistral/Vistral-7B-Chat"
+    # Lấy token từ environment variable
+    hf_token = os.getenv("HF_AUTH_TOKEN")
+    model = AutoModelForCausalLM.from_pretrained(model_name, token=hf_token)
+    tokenizer = AutoTokenizer.from_pretrained(model_name, token=hf_token)
     return model, tokenizer
 
 # Setup and load the model