Add HF token login

app.py

@@ -1,11 +1,20 @@
-from fastapi import FastAPI, Request
+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")  
+model = SentenceTransformer("mental/mental-bert-base-uncased")
 
 class InputText(BaseModel):
     text: str