Update app.py

app.py

@@ -64,11 +64,10 @@ def run_inference(image, device, valid_ds):
 # Create a Gradio interface
 iface = gr.Interface(
     fn=lambda image: run_inference(image, device, valid_ds),
-    inputs=gr.inputs.Image(type="numpy"),  # Input is an image
+    inputs=gr.Image(type="numpy"),  # Updated to use gr.Image
     outputs="text",  # Output is text (predicted class)
     title="Image Classification",
     description="Upload an image to get the predicted class using the ViT model."
 )
-
 # Launch the Gradio app
 iface.launch()