DHEIVER commited on
Commit
63dbfda
·
1 Parent(s): b1d7beb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ def classify_image(image_file):
36
  iface = gr.Interface(
37
  fn=classify_image,
38
  inputs=gr.inputs.Image(type="filepath", label="Upload an image"),
39
- outputs=[gr.outputs.Image(), "text"],
40
  title="Image Classifier",
41
  description="Classify images using a pre-trained ViT model",
42
  )
 
36
  iface = gr.Interface(
37
  fn=classify_image,
38
  inputs=gr.inputs.Image(type="filepath", label="Upload an image"),
39
+ outputs=[gr.outputs.Image(type="numpy"), "text"],
40
  title="Image Classifier",
41
  description="Classify images using a pre-trained ViT model",
42
  )