CK42 commited on
Commit
46e55ed
·
1 Parent(s): 0da04c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -47,12 +47,15 @@ def get_prediction(model_id):
47
 
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 [x for x in prediction]
56
  return predict
57
 
58
  with app:
 
47
 
48
  def predict(review):
49
  prediction = classifier(review)
50
+ predictions = []
51
+ for p in prediction:
52
+ new_pred = print(prediction)
53
+ predictions.append(new_pred)
54
  # stars = prediction[0]['label']
55
  # stars = (int)(stars.split('_')[1])+1
56
  # score = 100*prediction[0]['score']
57
  # return "{} {:.0f}%".format("\U00002B50"*stars, score)
58
+ return predictions
59
  return predict
60
 
61
  with app: