Add rounding to the score
Browse files
app.py
CHANGED
@@ -40,7 +40,7 @@ def classify_steam_review(input_text):
|
|
40 |
result = classifier(input_text)
|
41 |
|
42 |
label_1, label_2 = result[0][0]["label"], result[0][1]["label"]
|
43 |
-
score_1, score_2 = result[0][0]["score"], result[0][1]["score"]
|
44 |
|
45 |
return {"label_1": label_1, "score_1": score_1, "label_2": label_2, "score_2": score_2}
|
46 |
|
|
|
40 |
result = classifier(input_text)
|
41 |
|
42 |
label_1, label_2 = result[0][0]["label"], result[0][1]["label"]
|
43 |
+
score_1, score_2 = round(result[0][0]["score"], 6), round(result[0][1]["score"], 6)
|
44 |
|
45 |
return {"label_1": label_1, "score_1": score_1, "label_2": label_2, "score_2": score_2}
|
46 |
|