CK42 commited on
Commit
35fc62d
·
1 Parent(s): db66b47

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -48,10 +48,11 @@ def get_prediction(model_id):
48
  def predict(review):
49
  prediction = classifier(review)
50
  print(prediction)
51
- stars = prediction[0]['label']
52
- stars = (int)(stars.split('_')[1])+1
53
- score = 100*prediction[0]['score']
54
- return "{} {:.0f}%".format("\U00002B50"*stars, score)
 
55
  return predict
56
 
57
  with app:
 
48
  def predict(review):
49
  prediction = classifier(review)
50
  print(prediction)
51
+ # stars = prediction[0]['label']
52
+ # stars = (int)(stars.split('_')[1])+1
53
+ # score = 100*prediction[0]['score']
54
+ # return "{} {:.0f}%".format("\U00002B50"*stars, score)
55
+ return prediction
56
  return predict
57
 
58
  with app: