bohmian commited on
Commit
30d845d
·
verified ·
1 Parent(s): 558b359

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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"chroma_db/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)
 
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)