cmagganas commited on
Commit
4ec0045
·
1 Parent(s): 62047b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -88,13 +88,17 @@ async def init():
88
  embedder = CacheBackedEmbeddings.from_bytes_store(core_embeddings_model, store, namespace=core_embeddings_model.model)
89
 
90
  # set up FAISS vector store for csv
91
- barbie_csv_faiss_retriever = await cl.make_async(FAISS.from_documents)(chunked_barbie_csv_docs, embedder) #.as_retriever()
92
- opp_csv_faiss_retriever = await cl.make_async(FAISS.from_documents)(chunked_opp_csv_docs, embedder) #.as_retriever()
 
 
93
 
94
  # set up BM25 retriever
95
- barbie_wikipedia_bm25_retriever = await cl.make_async(BM25Retriever.from_documents)(chunked_barbie_wikipedia_docs)
 
96
  barbie_wikipedia_bm25_retriever.k = 1
97
- opp_wikipedia_bm25_retriever = await cl.make_async(BM25Retriever.from_documents)(chunked_opp_wikipedia_docs)
 
98
  opp_wikipedia_bm25_retriever.k = 1
99
 
100
  # set up FAISS vector store for Wiki
 
88
  embedder = CacheBackedEmbeddings.from_bytes_store(core_embeddings_model, store, namespace=core_embeddings_model.model)
89
 
90
  # set up FAISS vector store for csv
91
+ barbie_csv_faiss_async = await cl.make_async(FAISS.from_documents)(chunked_barbie_csv_docs, embedder)
92
+ barbie_csv_faiss_retriever = barbie_csv_faiss_async.as_retriever()
93
+ opp_csv_faiss_async = await cl.make_async(FAISS.from_documents)(chunked_opp_csv_docs, embedder)
94
+ opp_csv_faiss_retriever = opp_csv_faiss_async.as_retriever()
95
 
96
  # set up BM25 retriever
97
+ barbie_wikipedia_bm25_async = await cl.make_async(BM25Retriever.from_documents)(chunked_barbie_wikipedia_docs)
98
+ barbie_wikipedia_bm25_retriever = barbie_wikipedia_bm25_async.as_retriever()
99
  barbie_wikipedia_bm25_retriever.k = 1
100
+ opp_wikipedia_bm25_async = await cl.make_async(BM25Retriever.from_documents)(chunked_opp_wikipedia_docs)
101
+ opp_wikipedia_bm25_retriever = opp_wikipedia_bm25_async.as_retriever()
102
  opp_wikipedia_bm25_retriever.k = 1
103
 
104
  # set up FAISS vector store for Wiki