Update main.py

main.py

@@ -33,13 +33,17 @@ HF_TOKEN = os.getenv("HF_TOKEN")  # Set in environment variables
 FAISS_INDEX_PATH = snapshot_download(
     repo_id="negi2725/dataRag",
     repo_type="dataset",
-    token=HF_TOKEN
+    token=HF_TOKEN,
+    local_dir="./hf_cache/faiss_index",
+    local_dir_use_symlinks=False
 )
 
 # Download LegalBERT model
 MODEL_PATH = snapshot_download(
     repo_id="negi2725/legalBert",
-    token=HF_TOKEN
+    token=HF_TOKEN,
+    local_dir="./hf_cache/legalbert_model",
+    local_dir_use_symlinks=False
 )
 
 logger.info(f"FAISS index files downloaded to: {FAISS_INDEX_PATH}")