Add multi-label support

app.py

@@ -38,9 +38,13 @@ for label in labels:
 
 
 def predict(text):
-    preds = {
-        label: pipes[label](text)[0][1]['score'] for label in labels
-    }
+    preds = {}
+
+    for label in labels:
+        pred_array = pipes[label](text)[0]
+        for pred in pred_array:
+            if pred['label'] == 'LABEL_1':
+                preds[label] = pred['score']
 
     return preds