Spaces:
Sleeping
Sleeping
NCTCMumbai
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -35,8 +35,8 @@ 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 = ['My transhipment cargo is missing',
|
42 |
'What are benefits of the AEO Scheme and eligibility criteria?',
|
@@ -143,7 +143,7 @@ with gr.Blocks() as demo:
|
|
143 |
)
|
144 |
txt_btn = gr.Button(value="Submit text", scale=1)
|
145 |
|
146 |
-
api_kind = gr.Radio(choices=["HuggingFace"
|
147 |
|
148 |
prompt_html = gr.HTML()
|
149 |
try:
|
|
|
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 = ['My transhipment cargo is missing',
|
42 |
'What are benefits of the AEO Scheme and eligibility criteria?',
|
|
|
143 |
)
|
144 |
txt_btn = gr.Button(value="Submit text", scale=1)
|
145 |
|
146 |
+
api_kind = gr.Radio(choices=["HuggingFace"], value="HuggingFace")
|
147 |
|
148 |
prompt_html = gr.HTML()
|
149 |
try:
|