Christopher Román Jaimes commited on
Commit
c5c7b6b
·
1 Parent(s): 40d9906

chore: update BERT threshold.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -264,7 +264,7 @@ label_names_dict = {
264
  'LABEL_2': 2,
265
  'LABEL_3': 3,
266
  }
267
- BERT_SCORE_LIMIT = 0.98
268
 
269
  def extract_max_label_score(probabilities):
270
  # Find the dictionary with the maximum score
 
264
  'LABEL_2': 2,
265
  'LABEL_3': 3,
266
  }
267
+ BERT_SCORE_LIMIT = 0.980819808198082,
268
 
269
  def extract_max_label_score(probabilities):
270
  # Find the dictionary with the maximum score