Update main.py

main.py

@@ -6,10 +6,15 @@ import requests as req
 import pandas as pd
 import chromadb
 from sentence_transformers import SentenceTransformer
-from huggingface_hub import snapshot_download
+from huggingface_hub import snapshot_download , login
 import os
 from dotenv import load_dotenv
 
+import os
+
+HF_TOKEN = os.getenv("HF_TOKEN")
+login(token=HF_TOKEN)
+
 GROQ_API_KEY =  os.getenv("GROQ_API_KEY")
 GROQ_URL       = "https://api.groq.com/openai/v1/chat/completions"
 GROQ_MODEL     = "llama-3.1-8b-instant"
@@ -23,7 +28,8 @@ if not os.path.exists(CHROMA_DIR):
     snapshot_download(
         repo_id="Mohammedelhakim/parenting-qa-vectordb",
         repo_type="dataset",
-        local_dir="chroma_db"
+        local_dir="chroma_db",
+        token=HF_TOKEN
        
 )
     print("Vector database downloaded!")