Update app.py

app.py

@@ -5,24 +5,28 @@ import os
 
 MODEL_PATH = "hariharan220/finbert-stock-sentiment"
 
-# ✅ Load Hugging Face API token from environment variable
-HF_API_TOKEN = os.getenv("HF_API_KEY")
-if not HF_API_TOKEN:
+# ✅ Load Hugging Face API Key from environment variables
+HF_API_KEY = os.getenv("HF_API_KEY")
+if not HF_API_KEY:
     raise ValueError("❌ ERROR: Hugging Face API Key (HF_API_KEY) is missing. Set it in your environment variables.")
 
-# ✅ Authenticate when loading model
-print("🔍 Authenticating with Hugging Face...")
-os.system(f"huggingface-cli login --token {HF_API_TOKEN}")
+# ✅ Configure authentication for Hugging Face
+os.environ["HF_HOME"] = "/root/.cache/huggingface"
+os.environ["HF_API_TOKEN"] = HF_API_KEY  # Explicitly set API key
+os.environ["HUGGINGFACE_HUB_TOKEN"] = HF_API_KEY
 
-print("🔄 Loading Model...")
+print("🔄 Authenticating with Hugging Face...")
+
+# ✅ Load the model with authentication
 model = BertForSequenceClassification.from_pretrained(
     MODEL_PATH, 
-    token=HF_API_TOKEN  # Explicit authentication
+    token=HF_API_KEY  # Explicit authentication
 )
 tokenizer = AutoTokenizer.from_pretrained(
     MODEL_PATH, 
-    token=HF_API_TOKEN  # Explicit authentication
+    token=HF_API_KEY  # Explicit authentication
 )
+
 print("✅ Model Loaded Successfully!")
 
 # ✅ Define sentiment labels