Commit ·
32edcc4
1
Parent(s): 4400012
Update app.py
Browse files
app.py
CHANGED
|
@@ -35,12 +35,12 @@ template = env.get_template('template.j2')
|
|
| 35 |
template_html = env.get_template('template_html.j2')
|
| 36 |
|
| 37 |
# crossEncoder
|
| 38 |
-
cross_encoder = CrossEncoder('cross-encoder/ms-marco-MiniLM-L-6-v2')
|
| 39 |
-
|
| 40 |
# Examples
|
| 41 |
-
examples = ['What is the
|
| 42 |
-
'
|
| 43 |
-
|
| 44 |
|
| 45 |
|
| 46 |
def add_text(history, text):
|
|
@@ -51,7 +51,7 @@ def add_text(history, text):
|
|
| 51 |
|
| 52 |
def bot(history, api_kind):
|
| 53 |
top_rerank = 15
|
| 54 |
-
top_k_rank =
|
| 55 |
query = history[-1][0]
|
| 56 |
|
| 57 |
if not query:
|
|
|
|
| 35 |
template_html = env.get_template('template_html.j2')
|
| 36 |
|
| 37 |
# crossEncoder
|
| 38 |
+
#cross_encoder = CrossEncoder('cross-encoder/ms-marco-MiniLM-L-6-v2')
|
| 39 |
+
cross_encoder = CrossEncoder('BAAI/bge-reranker-base')
|
| 40 |
# Examples
|
| 41 |
+
examples = ['What is the 4 digit classification heading for Gold jewellery?',
|
| 42 |
+
'What is the 6 digit classification heading for Mobile phone?',
|
| 43 |
+
]
|
| 44 |
|
| 45 |
|
| 46 |
def add_text(history, text):
|
|
|
|
| 51 |
|
| 52 |
def bot(history, api_kind):
|
| 53 |
top_rerank = 15
|
| 54 |
+
top_k_rank = 10
|
| 55 |
query = history[-1][0]
|
| 56 |
|
| 57 |
if not query:
|