Update app.py

app.py

@@ -22,6 +22,18 @@ def setup_logging():
 
 logger = setup_logging()
 
+try:
+    token = os.environ.get('HF_TOKEN')
+    if not token:
+        logger.error("No HF_TOKEN found in environment variables")
+        raise ValueError("HF_TOKEN not found")
+    
+    login(token=token)
+    logger.info("Successfully logged in to Hugging Face")
+except Exception as e:
+    logger.error(f"Failed to login to Hugging Face: {e}")
+    raise
+
 class EnhancedTokenizer:
     def __init__(self, vocab):
         self.vocab = vocab
@@ -99,19 +111,19 @@ class ModelInference:
 
     def load_model(self):
         try:
-            # Download model and tokenizer
+            # Download model and tokenizer using authenticated session
             self.logger.info(f"Downloading from {MODEL_REPO}")
             
             model_path = hf_hub_download(
                 repo_id=MODEL_REPO,
                 filename="model.pt",
-                token=True  # Will use token if logged in
+                token=token  # Use token from environment
             )
             
             tokenizer_path = hf_hub_download(
                 repo_id=MODEL_REPO,
                 filename="tokenizer.json",
-                token=True
+                token=token  # Use token from environment
             )
             
             # Load model data