CK42 commited on
Commit
258c1c8
1 Parent(s): 534efde

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -22,9 +22,7 @@ def get_prediction(model_id):
22
  def predict(review):
23
  prediction = classifier(review)
24
  print(prediction)
25
- label = prediction[]['label']
26
- score = 100*prediction[]['score']
27
- return (label, score)
28
  return predict
29
 
30
  with app:
 
22
  def predict(review):
23
  prediction = classifier(review)
24
  print(prediction)
25
+ return list(prediction)
 
 
26
  return predict
27
 
28
  with app: