Update app.py

app.py

@@ -2,16 +2,23 @@ from fastapi import FastAPI
 from transformers import BertForSequenceClassification, AutoTokenizer
 import torch
 import os
-# ✅ Ensure HF_API_KEY is set
-HF_TOKEN = os.getenv("HF_API_KEY")
 
-if not HF_TOKEN:
-    raise ValueError("❌ HF_API_KEY is missing! Set it in Hugging Face Spaces Secrets.")
-
-# ✅ Load trained FinBERT model
 MODEL_PATH = "hariharan220/finbert-stock-sentiment"
-model = BertForSequenceClassification.from_pretrained(MODEL_PATH)
-tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
+
+# ✅ Load Hugging Face API token from environment variable
+HF_API_TOKEN = os.getenv("HF_API_KEY")  # Change from HF_API_TOKEN to HF_API_KEY
+if not HF_API_TOKEN:
+    raise ValueError("❌ ERROR: Hugging Face API Key (HF_API_KEY) is missing. Set it in your environment variables.")
+
+# ✅ Authenticate when loading model
+model = BertForSequenceClassification.from_pretrained(
+    MODEL_PATH, 
+    use_auth_token=HF_API_TOKEN
+)
+tokenizer = AutoTokenizer.from_pretrained(
+    MODEL_PATH, 
+    use_auth_token=HF_API_TOKEN
+)
 
 # ✅ Define sentiment labels
 labels = ["Negative", "Neutral", "Positive"]