Update app.py

app.py

@@ -1,6 +1,5 @@
 from transformers import ViTImageProcessor, ViTForImageClassification
 from PIL import Image
-import requests
 import gradio as gr
 
 # Load the model and processor
@@ -21,11 +20,12 @@ def classify_image(image):
 
 iface = gr.Interface(
     fn=classify_image,
-    inputs=gr.inputs.Image(type="numpy", label="Upload an Image"),
-    outputs=gr.outputs.Textbox(label="Predicted Class"),
+    inputs=gr.Image(type="numpy", label="Upload an Image"),
+    outputs=gr.Textbox(label="Predicted Class"),
     title="Image Classification with ViT",
     description="Upload an image to classify it using the Vision Transformer (ViT) model."
 )
 
 if __name__ == "__main__":
     iface.launch()
+