Update app.py
Browse files
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 |
)
|