HarryLee commited on
Commit
1dc9db2
·
1 Parent(s): 8533076

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -101,6 +101,7 @@ bm25 = BM25Okapi(tokenized_corpus)
101
  def word_len(s):
102
  return len([i for i in s.split(' ') if i])
103
 
 
104
  # This function will search all wikipedia articles for passages that
105
  # answer the query
106
  def search(query):
@@ -186,7 +187,7 @@ def search(query):
186
  else:
187
  res_clean.append(out)
188
 
189
- show_out = []
190
  for i in res_clean:
191
  num = word_len(i)
192
  if num > 1:
@@ -196,7 +197,16 @@ def search(query):
196
  for i in show_out[0:maxtags_sidebar]:
197
  st.write(i)
198
 
 
 
 
 
 
199
  st.write("## Results:")
200
  if st.button('Generated Expansion'):
201
  out_res = search(query = user_query)
 
 
 
 
202
  #st.success(out_res)
 
101
  def word_len(s):
102
  return len([i for i in s.split(' ') if i])
103
 
104
+ show_out = []
105
  # This function will search all wikipedia articles for passages that
106
  # answer the query
107
  def search(query):
 
187
  else:
188
  res_clean.append(out)
189
 
190
+
191
  for i in res_clean:
192
  num = word_len(i)
193
  if num > 1:
 
197
  for i in show_out[0:maxtags_sidebar]:
198
  st.write(i)
199
 
200
+ def reranking:
201
+ random.shuffle(show_out[0:maxtags_sidebar])
202
+ for i in show_out[0:maxtags_sidebar]:
203
+ st.write(i)
204
+
205
  st.write("## Results:")
206
  if st.button('Generated Expansion'):
207
  out_res = search(query = user_query)
208
+ #st.success(out_res)
209
+
210
+ if st.button('Rerank'):
211
+ out_res = reranking()
212
  #st.success(out_res)