abullard1 commited on
Commit
f486d2a
1 Parent(s): a7d7f3e

Add rounding to the score

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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