Update main.py

main.py

@@ -20,6 +20,9 @@ HF_TOKEN = os.getenv("HF_TOKEN")
 if not HF_TOKEN:
     raise RuntimeError("Hugging Face token is missing! Please set the HF_TOKEN environment variable.")
 
+# Set the Hugging Face token in the environment variable
+os.environ["HUGGINGFACE_HUB_TOKEN"] = HF_TOKEN
+
 app = FastAPI()
 
 # Model names
@@ -31,15 +34,13 @@ ENGLISH_MODEL_NAME = "siebert/sentiment-roberta-large-english"
 try:
     multilingual_tokenizer = AutoTokenizer.from_pretrained(
         MULTILINGUAL_TOKENIZER_NAME,
-        token=HF_TOKEN,  # Pass token here
         cache_dir=cache_dir
     )
 
     multilingual_model = pipeline(
         "sentiment-analysis",
         model=MULTILINGUAL_MODEL_NAME,
-        tokenizer=multilingual_tokenizer,
-        token=HF_TOKEN  # Pass token here
+        tokenizer=multilingual_tokenizer
     )
 except Exception as e:
     raise RuntimeError(f"Failed to load multilingual model: {e}")
@@ -48,8 +49,7 @@ except Exception as e:
 try:
     english_model = pipeline(
         "sentiment-analysis",
-        model=ENGLISH_MODEL_NAME,
-        token=HF_TOKEN  # Pass token here
+        model=ENGLISH_MODEL_NAME
     )
 except Exception as e:
     raise RuntimeError(f"Failed to load English sentiment model: {e}")