Update app.py

app.py

@@ -25,10 +25,10 @@ def predict(image):
     
     return f"Predicted class: {predicted_class_label}"
 
-# Create Gradio Interface
+# Create Gradio Interface (Note the change here: `gr.Image` and `gr.Text`)
 interface = gr.Interface(fn=predict, 
-                         inputs=gr.inputs.Image(type="pil", label="Upload Image"),
-                         outputs="text", 
+                         inputs=gr.Image(type="pil", label="Upload Image"),
+                         outputs=gr.Text(), 
                          title="ViT Image Classification (ImageNet)", 
                          description="Upload an image to classify it into one of the 1000 ImageNet classes.")