Update src/streamlit_app.py

src/streamlit_app.py

@@ -113,22 +113,19 @@ model = CLIPModel.from_pretrained(
 ).to("cpu").eval()
 
 
-# Load FAISS index + id_map
-index_path = hf_hub_download(
-    repo_id=model_repo,
-    filename="faiss_index.bin",
-    token=hf_token
-)
-mapping_path = hf_hub_download(
-    repo_id=model_repo,
-    filename="id_map.json",
-    token=hf_token
-)
+index_path   = os.path.join(model_dir, "faiss_index.bin")
+mapping_path = os.path.join(model_dir, "id_map.json")
+
+# Kiểm tra file tồn tại
+assert os.path.isfile(index_path),   f"Không tìm thấy {index_path}"
+assert os.path.isfile(mapping_path), f"Không tìm thấy {mapping_path}"
+
+# Load index
 index = faiss.read_index(index_path)
 
-# Load processor
+# 4. Load processor cũng từ thư mục local
 processor = CLIPProcessor.from_pretrained(
-    model_repo,
+    model_dir,
     use_auth_token=hf_token
 )