svjack commited on
Commit
f9e3d75
·
verified ·
1 Parent(s): 41041aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -441,7 +441,9 @@ texts = book_df["content_chunks_formatted"].dropna().drop_duplicates().values.to
441
  #embedding_path = "bge-small-book-qa/"
442
  embedding_path = "svjack/bge-small-book-qa"
443
  bge_book_embeddings = HuggingFaceEmbeddings(model_name=embedding_path)
444
- docsearch_bge_loaded = FAISS.load_local("bge_small_book_chunks_prebuld/", bge_book_embeddings)
 
 
445
 
446
  from setfit import SetFitModel
447
  setfit_model = SetFitModel.from_pretrained("svjack/setfit_info_cls")
 
441
  #embedding_path = "bge-small-book-qa/"
442
  embedding_path = "svjack/bge-small-book-qa"
443
  bge_book_embeddings = HuggingFaceEmbeddings(model_name=embedding_path)
444
+ docsearch_bge_loaded = FAISS.load_local("bge_small_book_chunks_prebuld/", bge_book_embeddings,
445
+ allow_dangerous_deserialization = True
446
+ )
447
 
448
  from setfit import SetFitModel
449
  setfit_model = SetFitModel.from_pretrained("svjack/setfit_info_cls")