Jai12345 commited on
Commit
9c6136a
1 Parent(s): 8f9797b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -138,8 +138,6 @@ def cross_encode():
138
  return cross_encoder
139
 
140
 
141
- bi_enc_options = ["multi-qa-mpnet-base-dot-v1", "all-mpnet-base-v2", "multi-qa-MiniLM-L6-cos-v1"]
142
-
143
 
144
  def display_as_table(model, top_k=2, score='score'):
145
  # Display the df with text and scores as a table
@@ -154,7 +152,7 @@ def display_as_table(model, top_k=2, score='score'):
154
  st.title("Search with Retrieve & Rerank")
155
  window_size = 3
156
 
157
-
158
  # This function will search all wikipedia articles for passages that answer the query
159
  def search_func(query, top_k=2):
160
  global bi_encoder, cross_encoder
 
138
  return cross_encoder
139
 
140
 
 
 
141
 
142
  def display_as_table(model, top_k=2, score='score'):
143
  # Display the df with text and scores as a table
 
152
  st.title("Search with Retrieve & Rerank")
153
  window_size = 3
154
 
155
+ bi_encoder_type="multi-qa-mpnet-base-dot-v1"
156
  # This function will search all wikipedia articles for passages that answer the query
157
  def search_func(query, top_k=2):
158
  global bi_encoder, cross_encoder