added quotes to pixel_values

app.py

@@ -15,7 +15,7 @@ labels = dataset['train'].features['labels'].names
 
 def classify(im):
   features = feature_extractor(im, return_tensors='pt')
-  logits = model(features[pixel_values])[-1]
+  logits = model(features["pixel_values"])[-1]
   probability = torch.nn.functional.softmax(logits, dim=-1)
   probs = probability[0].detach().numpy()
   confidences = {label: float(probs[i]) for i, label in enumerate(labels)}