kapllan commited on
Commit
295728b
·
verified ·
1 Parent(s): 671198b

Rounding the scores

Browse files

The scores are now shorter because now we round them.

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -70,7 +70,7 @@ def predict_topic(text: str) -> [List[str], str]:
70
  continue
71
  feature_values = [(col, row[0, col]) for col in row.nonzero()[1]]
72
  for subj_id, score in model.predict(feature_values, top_k=1000):
73
- results.append((id2label[str(subj_id)], score))
74
  return results, language
75
 
76
 
 
70
  continue
71
  feature_values = [(col, row[0, col]) for col in row.nonzero()[1]]
72
  for subj_id, score in model.predict(feature_values, top_k=1000):
73
+ results.append((id2label[str(subj_id)], round(score,2)))
74
  return results, language
75
 
76