Update app.py

app.py

@@ -1,15 +1,14 @@
-
-from  datasets import load_dataset
 import gradio as gr
 import torch
-from transformers import AutoFeatureExtractor, AutoModelForImageClassification
+from transformers import AutoFeatureExtractor, AutoModelForImageClassification, ViTImageProcessor
 
-dataset = load_dataset("beans") 
+image_processor = ViTImageProcessor.from_pretrained("google/vit-base-patch16-224")
 
 extractor = AutoFeatureExtractor.from_pretrained("saved_model_files")
 model = AutoModelForImageClassification.from_pretrained("saved_model_files")
 
-labels = dataset['train'].features['labels'].names
+labels = ['angular_leaf_spot', 'bean_rust', 'healthy'] 
+
 
 def classify(im):
   features = image_processor(im, return_tensors='pt')