Update app.py
Browse files
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
|
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 |
""")
|