Update app.py

app.py

@@ -15,6 +15,15 @@ model_id_3 = "distilbert-base-uncased-finetuned-sst-2-english"
 model_id_4 = "lordtt13/emo-mobilebert"
 model_id_5 = "juliensimon/reviews-sentiment-analysis"
 
+def parse_output(output_json):  # list of list of dicts
+    list_pred=[]
+    for i in range(len(output_json[0])):
+        label = output_json[0][i]['label']
+        score = output_json[0][i]['score']
+        list_pred.append((label, score))
+    return list_pred
+
+
 def get_prediction(model_id):
 
     classifier = pipeline("text-classification", model=model_id, return_all_scores=True)
@@ -22,7 +31,7 @@ def get_prediction(model_id):
     def predict(review):
             prediction = classifier(review)
             print(prediction)
-            return prediction
+            return parse_output(prediction)
     return predict
 
 with app: