Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -260,7 +260,7 @@ bm25_new_retrievers = {} # to store retrievers for different countries
|
|
260 |
# get retrievers for country which we override
|
261 |
if len(st.session_state['countries_override']) > 0:
|
262 |
for country in st.session_state['countries_override']:
|
263 |
-
chroma_db_new = Chroma(persist_directory=f"
|
264 |
bm25_filename = f"bm25/new_{country}_chunk_{st.session_state['chunk_size']}_overlap_{st.session_state['chunk_overlap']}_.pickle"
|
265 |
with open(bm25_filename, 'rb') as handle:
|
266 |
bm25_retriever = pickle.load(handle)
|
|
|
260 |
# get retrievers for country which we override
|
261 |
if len(st.session_state['countries_override']) > 0:
|
262 |
for country in st.session_state['countries_override']:
|
263 |
+
chroma_db_new = Chroma(persist_directory=f"chromadb/new_{country}_chunk_{st.session_state['chunk_size']}_overlap_{st.session_state['chunk_overlap']}_",embedding_function=hf_embeddings)
|
264 |
bm25_filename = f"bm25/new_{country}_chunk_{st.session_state['chunk_size']}_overlap_{st.session_state['chunk_overlap']}_.pickle"
|
265 |
with open(bm25_filename, 'rb') as handle:
|
266 |
bm25_retriever = pickle.load(handle)
|