NCTCMumbai commited on
Commit
502c6e6
1 Parent(s): c7c1263

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
- #cross_encoder = CrossEncoder('BAAI/bge-reranker-base')
40
  # Examples
41
  examples = ['My transhipment cargo is missing','can u explain and tabulate difference between b 17 bond and a warehousing bond',
42
  'What are benefits of the AEO Scheme and eligibility criteria?',
@@ -52,7 +52,7 @@ def add_text(history, text):
52
 
53
  def bot(history, api_kind):
54
  top_rerank = 15
55
- top_k_rank = 8
56
  query = history[-1][0]
57
  print('history[-1][0]',history[-1][0])
58
  print('api kind ',api_kind)
 
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','can u explain and tabulate difference between b 17 bond and a warehousing bond',
42
  'What are benefits of the AEO Scheme and eligibility criteria?',
 
52
 
53
  def bot(history, api_kind):
54
  top_rerank = 15
55
+ top_k_rank = 10
56
  query = history[-1][0]
57
  print('history[-1][0]',history[-1][0])
58
  print('api kind ',api_kind)