Update app.py

app.py

@@ -1,13 +1,11 @@
 import gradio as gr
-from transformers import ViTForImageClassification, ViTImageProcessor
+from transformers import ViTForImageClassification, AutoImageProcessor
 from PIL import Image
 import torch
 
-# মডেল ও প্রসেসর লোড করো
 model = ViTForImageClassification.from_pretrained("rakib730/vit-base-oxford-iiit-pets")
-processor = ViTImageProcessor.from_pretrained("rakib730/vit-base-oxford-iiit-pets")
+processor = AutoImageProcessor.from_pretrained("rakib730/vit-base-oxford-iiit-pets")
 
-# প্রেডিকশন ফাংশন
 def classify_image(image):
     inputs = processor(images=image, return_tensors="pt")
     with torch.no_grad():
@@ -17,7 +15,6 @@ def classify_image(image):
     label = model.config.id2label[predicted_class_idx]
     return label
 
-# Gradio UI
 iface = gr.Interface(
     fn=classify_image,
     inputs=gr.Image(type="pil"),