Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import torch
-from transformers import AutoFeatureExtractor, AutoModelForImageClassification, ViTImageProcessor
+from transformers import ViTImageProcessor, AutoFeatureExtractor, AutoModelForImageClassification
 
 image_processor = ViTImageProcessor.from_pretrained("google/vit-base-patch16-224")
 
@@ -10,8 +10,8 @@ model = AutoModelForImageClassification.from_pretrained("saved_model_files")
 labels = ['angular_leaf_spot', 'bean_rust', 'healthy'] 
 
 
-def classify(im):
-  features = image_processor(im, return_tensors='pt')
+def classify(image):
+  features = image_processor(image, return_tensors='pt')
   logits = model(features["pixel_values"])[-1]
   probability = torch.nn.functional.softmax(logits, dim=-1)
   probs = probability[0].detach().numpy()