svjack commited on
Commit
62a3f7f
1 Parent(s): 0329fc1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -429,7 +429,9 @@ texts = book_df["content_chunks_formatted"].dropna().drop_duplicates().values.to
429
  #embedding_path = "bge-small-book-qa/"
430
  embedding_path = "svjack/bge-small-book-qa"
431
  bge_book_embeddings = HuggingFaceEmbeddings(model_name=embedding_path)
432
- docsearch_bge_loaded = FAISS.load_local("bge_small_book_chunks_prebuld/", bge_book_embeddings)
 
 
433
 
434
  from setfit import SetFitModel
435
  #setfit_model = SetFitModel.from_pretrained("setfit_info_cls")
 
429
  #embedding_path = "bge-small-book-qa/"
430
  embedding_path = "svjack/bge-small-book-qa"
431
  bge_book_embeddings = HuggingFaceEmbeddings(model_name=embedding_path)
432
+ docsearch_bge_loaded = FAISS.load_local("bge_small_book_chunks_prebuld/", bge_book_embeddings,
433
+ allow_dangerous_deserialization = True
434
+ )
435
 
436
  from setfit import SetFitModel
437
  #setfit_model = SetFitModel.from_pretrained("setfit_info_cls")