Update app.py

app.py

@@ -7,7 +7,7 @@ import torch
 from sentence_transformers import CrossEncoder
 import gradio as gr
 from datetime import datetime 
-from huggingface_hub import hf_hub_download
+from huggingface_hub import hf_hub_download, HfApi
 
 # Load environment variables and initialize clients
 load_dotenv()
@@ -15,11 +15,14 @@ GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 client = Groq(api_key=GROQ_API_KEY)
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
+# Get the token from environment variables
+hf_token = os.getenv("HF_TOKEN")
+
 def load_chroma_db():
     # Load from Hugging Face dataset
     print("Using ChromaDB from Hugging Face dataset...")
-    # Download the database file from Hugging Face
-    db_file_path = hf_hub_download(repo_id="Mr-Geo/chroma_db", filename="chroma.sqlite3")
+    # Download the database file from Hugging Face with authentication
+    db_file_path = hf_hub_download(repo_id="Mr-Geo/chroma_db", filename="chroma.sqlite3", use_auth_token=hf_token)
     # Initialize your ChromaDB with the downloaded file
     db = chromadb.Client(path=db_file_path)
     return db