Update app.py

app.py

@@ -43,11 +43,12 @@ def classify_image_with_overlay(img):
 iface = gr.Interface(
     fn=classify_image_with_overlay,
     inputs=gr.inputs.Image(),
-    outputs=gr.outputs.Image(),
+    outputs=gr.outputs.Image(type="numpy"),
     live=True,
     title="ViT Image Classifier with Overlay",
     description="Upload an image for classification with label overlay.",
 )
 
+
 if __name__ == "__main__":
     iface.launch()