Update app.py

app.py

@@ -43,7 +43,7 @@ def get_metadata(model_id):
 
 def get_prediction(model_id):
 
-    classifier = pipeline("text-classification", model=model_id)
+    classifier = pipeline("text-classification", model=model_id, return_all_scores=True)
              
     def predict(review):
             prediction = classifier(review)
@@ -52,7 +52,7 @@ def get_prediction(model_id):
             # stars = (int)(stars.split('_')[1])+1
             # score = 100*prediction[0]['score']
             # return "{} {:.0f}%".format("\U00002B50"*stars, score)
-            # return print(prediction)
+            return prediction
     return predict
 
 with app: