Update app.py

app.py

@@ -14,14 +14,14 @@ import pickle
 ## Main page
 ############
 
-st.write("# Code for Query Expansion")
+st.write("# Demonstration for Etsy Query Expansion（Etsy-QE）")
 
 st.markdown("***Idea is to build a model which will take query as inputs and generate expansion information as outputs.***")
 image = Image.open('top.png')
 st.image(image)
 
-#st.sidebar.write("# Parameter Selection")
-#maxtags_sidebar = st.sidebar.slider('Number of query allowed?', 1, 10, 1, key='ehikwegrjifbwreuk')
+st.sidebar.write("# Top-N Selection")
+maxtags_sidebar = st.sidebar.slider('Number of query allowed?', 1, 20, 1, key='ehikwegrjifbwreuk')
 #user_query = st_tags(
 #    label='# Enter Query:',
 #    text='Press enter to add more',
@@ -83,21 +83,21 @@ def search(query):
     for idx in range(len(cross_scores)):
         hits[idx]['cross-score'] = cross_scores[idx]
 
-    # Output of top-10 hits from bi-encoder
+    # Output of top-N hits from bi-encoder
     st.write("\n-------------------------\n")
-    st.write("Top-10 Bi-Encoder Retrieval hits")
+    st.write("Top-N Bi-Encoder Retrieval hits")
     hits = sorted(hits, key=lambda x: x['score'], reverse=True)
-    for hit in hits[0:10]:
+    for hit in hits[0:maxtags_sidebar]:
         st.write("\t{:.3f}\t{}".format(hit['score'], passages[hit['corpus_id']].replace("\n", " ")))
 
-    # Output of top-10 hits from re-ranker
+    # Output of top-N hits from re-ranker
     st.write("\n-------------------------\n")
-    st.write("Top-10 Cross-Encoder Re-ranker hits")
+    st.write("Top-N Cross-Encoder Re-ranker hits")
     hits = sorted(hits, key=lambda x: x['cross-score'], reverse=True)
-    for hit in hits[0:10]:
+    for hit in hits[0:maxtags_sidebar]:
         st.write("\t{:.3f}\t{}".format(hit['cross-score'], passages[hit['corpus_id']].replace("\n", " ")))
 
 st.write("## Results:")
-if st.button('Generate Sentence'):
+if st.button('Generated Expansion'):
     out = search(query = user_query)
-    #st.success(out)
+    st.success(out)