Update handler.py

handler.py

@@ -84,7 +84,9 @@ class EndpointHandler:
         # run normal prediction
         scores = self.model.predict_proba(inputs)[0]
 
-        return [
+        results = [
             {"label": self.id2label[i], "score": score.item()}
             for i, score in enumerate(scores)
         ]
+        max_element = max(results, key=lambda x: x['score'])
+        return max_element