Update app.py

app.py

@@ -6,10 +6,10 @@ feature_extractor = AutoFeatureExtractor.from_pretrained("microsoft/dit-base-fin
 model = AutoModelForImageClassification.from_pretrained("microsoft/dit-base-finetuned-rvlcdip")
 
 def classify_image(image):
-   pixel_values = feature_extractor(image, return_tensors="pt")
+   encoding = feature_extractor(image, return_tensors="pt")
    
    with torch.no_grad():
-     outputs = model(pixel_values)
+     outputs = model(**encoding)
      logits = outputs.logits
      
    predicted_class = model.config.id2label[logits.argmax(-1).item()]