Update app.py

app.py

@@ -25,7 +25,7 @@ labels = []
 def predict(img):
     img = PILImage.create(img)
 
-    inputs = feature_extractor(images=images, return_tensors="pt")
+    inputs = feature_extractor(images=img, return_tensors="pt")
     logits = model(**inputs)['logits'].cpu().detach().numpy()
     predictions = np.argmax(logits, axis=1) 
     predicted_labels = [labels[prediction] for prediction in predictions]