Update app.py

app.py

@@ -8,7 +8,7 @@ model = AutoModelForImageClassification.from_pretrained("saved_model_files")
 labels = dataset['train'].features['labels'].names
 
 def classify(input_image):
-  features = image_processor(input_image, return_tensors='pt')
+  features = extractor(input_image, return_tensors='pt')
   logits = model(features["pixel_values"])[-1]
   probability = torch.nn.functional.softmax(logits, dim=-1)
   probs = probability[0].detach().numpy()