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