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'."