Update app.py

app.py

@@ -7,8 +7,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 def predict(image):
     classifier = pipeline(task="image-classification")
     preds = classifier(image)
-    preds = [{"score": round(pred["score"], 4), "label": pred["label"]} for pred in preds]
-    return [pred["label"] for pred in preds]  # Return a list of labels only
+    return {pred["label"]: float(round(pred["score"]), 4) for pred in preds}
 
 description = """
 """