Update handler.py

handler.py

@@ -84,7 +84,7 @@ class EndpointHandler:
         # run normal prediction
         scores = self.model.predict_proba(inputs)[0]
 
-        # get index of element with highest score
-        max_index = scores.argmax()
-
-        return {"label": self.id2label[max_index], "score": scores[max_index].item()}
+        return [
+            {"label": self.id2label[i], "score": score.item()}
+            for i, score in enumerate(scores)
+        ]