Update app.py

app.py

@@ -3,18 +3,17 @@ import os
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from huggingface_hub import login
 
-sec_token=os.getenv("HF_TOKEN")
-login(token=sec_token)
+login(token=os.getenv('HF_TOKEN'))
 
 import torch
 
 class VietnameseChatbot:
-    def __init__(self, model_name="meta-llama/Llama-2-13b-hf"):
+    def __init__(self, model_name="meta-llama/Llama-2-13b-hf", token=os.getenv('HF_TOKEN')):
         """
         Initialize the Vietnamese chatbot with a pre-trained model
         """
-        self.tokenizer = AutoTokenizer.from_pretrained(model_name, token=sec_token)
-        self.model = AutoModelForCausalLM.from_pretrained(model_name, token=sec_token)
+        self.tokenizer = AutoTokenizer.from_pretrained(model_name, token)
+        self.model = AutoModelForCausalLM.from_pretrained(model_name, token)
         
         # Use GPU if available
         self.device = "cuda" if torch.cuda.is_available() else "cpu"