NCTCMumbai commited on
Commit
ce11f6d
·
verified ·
1 Parent(s): fb15851

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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","Openai"], value="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: