Update app.py

app.py

@@ -29,7 +29,7 @@ transform = Compose([
 def classify_image(image):
     # Preprocess the image
     image = Image.fromarray(image)
-    inputs = processor(text=id2label.values(), images=image, return_tensors="pt", padding=True)
+    inputs = processor(text=labels, images=image, return_tensors="pt", padding=True)
 
     # Run inference
     outputs = model(**inputs)