Update app.py

app.py

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