fixed pred output

utils/inference.py

@@ -29,11 +29,5 @@ def query(filename):
     outputs = model(**inputs)
     logits_per_image = outputs.logits_per_image
     probs = logits_per_image.softmax(dim=1)
-    output = {
-        label: prob.item()
-        for label, prob in zip(
-            ["Happy", "Sad", "Surprised", "Angry", "Disgusted", "Neutral", "Fearful"],
-            probs[0],
-        )
-    }
+    output = [{"label": label, "score": float(score)} for label, score in zip(["Happy", "Sad", "Surprised", "Angry", "Disgusted", "Neutral", "Fearful"], probs[0])]
     return output