nightfury commited on
Commit
c1fa79c
·
verified ·
1 Parent(s): 7f3a89f

Update appChatbot.py

Browse files
Files changed (1) hide show
  1. 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"{book}_cache"
39
 
40
- vectorstore = snapshot_download(repo_id="calmgoose/book-embeddings",
41
  repo_type="dataset",
42
  revision="main",
43
- allow_patterns=f"books/{BOOK}/*", # to download only the one book
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