Update appChatbot.py
Browse files- appChatbot.py +3 -3
appChatbot.py
CHANGED
@@ -35,12 +35,12 @@ embeddings = HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")
|
|
35 |
ABS_PATH = os.path.dirname(os.path.abspath(__file__))
|
36 |
DB_DIR = os.path.join(ABS_PATH, "db")
|
37 |
|
38 |
-
cache_dir=f"
|
39 |
|
40 |
-
vectorstore = snapshot_download(repo_id="
|
41 |
repo_type="dataset",
|
42 |
revision="main",
|
43 |
-
allow_patterns=f"
|
44 |
cache_dir=cache_dir,
|
45 |
)
|
46 |
# get path to the `vectorstore` folder that you just downloaded
|
|
|
35 |
ABS_PATH = os.path.dirname(os.path.abspath(__file__))
|
36 |
DB_DIR = os.path.join(ABS_PATH, "db")
|
37 |
|
38 |
+
cache_dir=f"book_cache"
|
39 |
|
40 |
+
vectorstore = snapshot_download(repo_id="waterdb/book-embeddings",
|
41 |
repo_type="dataset",
|
42 |
revision="main",
|
43 |
+
allow_patterns=f"book/*", # to download only the one book
|
44 |
cache_dir=cache_dir,
|
45 |
)
|
46 |
# get path to the `vectorstore` folder that you just downloaded
|