Update app.py

app.py

@@ -1,5 +1,5 @@
 import torch
-from transformers import ViTFeatureExtractor, ViTForImageClassification
+from transformers import ViTImageProcessor, ViTForImageClassification
 from fastai.learner import load_learner
 from PIL import Image
 import matplotlib.pyplot as plt
@@ -10,7 +10,7 @@ import base64
 
 # 🔹 Cargar modelo ViT desde Hugging  Face
 MODEL_NAME = "ahishamm/vit-base-HAM-10000-sharpened-patch-32"
-feature_extractor = ViTFeatureExtractor.from_pretrained(MODEL_NAME)
+feature_extractor = ViTImageProcessor.from_pretrained(MODEL_NAME)
 model_vit = ViTForImageClassification.from_pretrained(MODEL_NAME)
 model_vit.eval()