nickmuchi commited on
Commit
609b6e0
1 Parent(s): 276eb3f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -211,7 +211,7 @@ top_k = st.sidebar.slider("Number of Top Hits Generated",min_value=1,max_value=5
211
 
212
  # This function will search all wikipedia articles for passages that
213
  # answer the query
214
- def search_func(query, top_k=top_k, bi_encoder_type):
215
 
216
  global bi_encoder, cross_encoder
217
 
@@ -359,7 +359,7 @@ if search:
359
  with st.spinner(
360
  text="Embedding completed, searching for relevant text for given query and hits..."):
361
 
362
- search_func(search_query,top_k)
363
 
364
  st.markdown("""
365
  """)
 
211
 
212
  # This function will search all wikipedia articles for passages that
213
  # answer the query
214
+ def search_func(query, bi_encoder_type, top_k=top_k):
215
 
216
  global bi_encoder, cross_encoder
217
 
 
359
  with st.spinner(
360
  text="Embedding completed, searching for relevant text for given query and hits..."):
361
 
362
+ search_func(search_query,bi_encoder_type,top_k)
363
 
364
  st.markdown("""
365
  """)