Spaces:
Running
Running
Rounding the scores
Browse filesThe scores are now shorter because now we round them.
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 |
|