Remove login(), pass token to SentenceTransformer

app.py

@@ -1,20 +1,16 @@
 from fastapi import FastAPI
 from pydantic import BaseModel
 from sentence_transformers import SentenceTransformer
-import torch
 import os
-from huggingface_hub import login
-
-# Log in using the HF_TOKEN environment variable
-hf_token = os.getenv("HF_TOKEN")
-if hf_token:
-    login(token=hf_token)
-else:
-    raise ValueError("HF_TOKEN not found. Make sure it's set in your Space variables.")
 
 app = FastAPI()
 
-model = SentenceTransformer("mental/mental-bert-base-uncased")
+hf_token = os.getenv("HF_TOKEN")
+if not hf_token:
+    raise ValueError("HF_TOKEN not set in environment variables.")
+
+# Pass token while loading the model
+model = SentenceTransformer("mental/mental-bert-base-uncased", use_auth_token=hf_token)
 
 class InputText(BaseModel):
     text: str