Update app.py

app.py

@@ -4,7 +4,8 @@ from transformers import pipeline
 def predict(image):
     model_id = "google/vit-base-patch16-224"
     classifier = pipeline("image-classification", model=model_id)
-    return classifier(image)
+    predictions = classifier(image)
+    return {prediction['label']: prediction['score'] for prediction in predictions}
 
 title = "Image Classifier"
 description = "A demo that recognizes and classifies images using the model from Hugging Face's 'google/vit-base-patch16-224'."