Christopher Román Jaimes
commited on
Commit
·
c5c7b6b
1
Parent(s):
40d9906
chore: update BERT threshold.
Browse files
app.py
CHANGED
@@ -264,7 +264,7 @@ label_names_dict = {
|
|
264 |
'LABEL_2': 2,
|
265 |
'LABEL_3': 3,
|
266 |
}
|
267 |
-
BERT_SCORE_LIMIT = 0.
|
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
|