Update app.py

app.py

@@ -3,7 +3,8 @@ import os
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from huggingface_hub import login
 
-login(token=os.environ['HF_TOKEN'])
+sec_token=os.getenv("HF_TOKEN")
+login(token=sec_token)
 
 import torch
 
@@ -12,8 +13,8 @@ class VietnameseChatbot:
         """
         Initialize the Vietnamese chatbot with a pre-trained model
         """
-        self.tokenizer = AutoTokenizer.from_pretrained(model_name)
-        self.model = AutoModelForCausalLM.from_pretrained(model_name)
+        self.tokenizer = AutoTokenizer.from_pretrained(model_name, token=sec_token)
+        self.model = AutoModelForCausalLM.from_pretrained(model_name, token=sec_token)
         
         # Use GPU if available
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -49,13 +50,9 @@ class VietnameseChatbot:
         return response
 
 def main():
-    st.set_page_config(
-            page_title="IOGPT", 
-        page_icon="🇻🇳",
-        layout="wide"
-    )
+    st.set_page_config(page_title="IOGPT", layout="wide")
 
-    st.title("🇻🇳 Chat với IOGPT")
+    st.title("Chat với IOGPT")
     st.markdown("""
     ### Trò chuyện với IOGPT
     """)