NCTCMumbai commited on
Commit
32edcc4
1 Parent(s): 4400012

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
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 capital of China?',
42
- 'Why is the sky blue?',
43
- 'Who won the mens world cup in 2014?', ]
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 = 4
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: