Update prepare_vector_dp.py

prepare_vector_dp.py

@@ -8,8 +8,11 @@ from huggingface_hub import hf_hub_download
 # from llama_cpp import Llama
 import os
 
+cache_path = "/home/user/app/hf_cache"
+os.makedirs(cache_path, exist_ok=True)
+
+#os.environ["HF_HOME"] = "/home/user/app/hf_cache"
 
-os.environ["HF_HOME"] = "/home/user/app/hf_cache"
 
 # Khai bao bien 
 pdf_data_path = "data"
@@ -17,7 +20,8 @@ vector_dp_path = "vectorstores/db_faiss"
 
 model_file = hf_hub_download(
     repo_id="Pudding48/TinyLlamaTest",  # 🟢 This must be a model repo, not a Space
-    filename="tinyllama-1.1b-chat-v1.0.Q8_0.gguf"
+    filename="tinyllama-1.1b-chat-v1.0.Q8_0.gguf",
+    cache_dir=cache_path
 )
 
 # Ham 1. Tao ra vector DB tu 1 doan text