nickmuchi commited on
Commit
87e6a8a
1 Parent(s): b92e86c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -162,7 +162,7 @@ def cross_encode():
162
 
163
  global cross_encoder
164
  #The bi-encoder will retrieve 100 documents. We use a cross-encoder, to re-rank the results list to improve the quality
165
- cross_encoder = CrossEncoder('BAAI/bge-reranker-base')
166
  return cross_encoder
167
 
168
  @st.experimental_memo(suppress_st_warning=True)
162
 
163
  global cross_encoder
164
  #The bi-encoder will retrieve 100 documents. We use a cross-encoder, to re-rank the results list to improve the quality
165
+ cross_encoder = CrossEncoder('cross-encoder/ms-marco-MiniLM-L-12-v2')
166
  return cross_encoder
167
 
168
  @st.experimental_memo(suppress_st_warning=True)