Update prepare_vector_dp.py

prepare_vector_dp.py

@@ -3,27 +3,25 @@ from langchain_community.document_loaders import PyPDFLoader, DirectoryLoader
 from langchain_community.vectorstores import FAISS
 from langchain_community.embeddings import GPT4AllEmbeddings
 
-#from huggingface_hub import hf_hub_download
+from huggingface_hub import hf_hub_download
 
-from llama_cpp import Llama
+# from llama_cpp import Llama
 import os
 
-model_file = Llama.from_pretrained(
-	repo_id="Pudding48/TinyLLamaTest",
-	filename="tinyllama-1.1b-chat-v1.0.Q8_0.gguf",
-)
+# model_file = Llama.from_pretrained(
+# 	repo_id="Pudding48/TinyLLamaTest",
+# 	filename="tinyllama-1.1b-chat-v1.0.Q8_0.gguf",
+# )
 
 # Khai bao bien 
 pdf_data_path = "data"
 vector_dp_path = "vectorstores/db_faiss"
 
-
-
-# model_file = hf_hub_download(
-#     repo_id="Pudding48/TinyLlamaTest",   # Replace with your model repo
-#     filename="tinyllama-1.1b-chat-v1.0.Q8_0.gguf",
-#     cache_dir="model"  # Will be created in the Space's environment
-# )
+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",
+    cache_dir="model"
+)
 
 # Ham 1. Tao ra vector DB tu 1 doan text
 def create_db_from_text():