hf token

src/streamlit_app.py

@@ -12,10 +12,21 @@ import pickle
 import chromadb
 from chromadb.utils import embedding_functions
 
-hf_token = os.getenv("HF_TOKEN")  # read from HuggingFace Space secret
+hf_token = os.getenv("HF_TOKEN")
+
 if hf_token:
-    os.environ["HF_TOKEN"] = hf_token
-    os.environ["HUGGINGFACE_HUB_TOKEN"] = hf_token
+    try:
+        # Use the official login function
+        login(token=hf_token) 
+        st.success("Hugging Face token successfully validated (or cached).")
+    except Exception as e:
+        # This might reveal a more specific error than AxiosError
+        st.error(f"Hugging Face login failed (check token validity): {e}")
+        # Still set the environment variables for downstream libraries
+        os.environ["HF_TOKEN"] = hf_token
+        os.environ["HUGGINGFACE_HUB_TOKEN"] = hf_token
+else:
+    st.warning("HF_TOKEN not found. Using anonymous access (may lead to 403 for private models).")
     
 BASE_DIR = "/tmp"
 os.makedirs(BASE_DIR, exist_ok=True)