Spaces:
Running
Running
Fix labels error
Browse files
app.py
CHANGED
@@ -29,7 +29,7 @@ def predict_image(image):
|
|
29 |
interface = gr.Interface(
|
30 |
fn=predict_image,
|
31 |
inputs=gr.Image(type="pil"),
|
32 |
-
outputs=[gr.Image(type="pil", label="Image Output"), label="
|
33 |
title="Animals Classifier",
|
34 |
description="Upload an image of an animal, and the model will predict it.\n\n**Disclaimer:** This model is trained only on specific animal classes (butterfly, cats, cow, dogs, elephant, horse, monkey, sheep, spider, squirrel) and may not accurately predict animals outside these classes."
|
35 |
)
|
|
|
29 |
interface = gr.Interface(
|
30 |
fn=predict_image,
|
31 |
inputs=gr.Image(type="pil"),
|
32 |
+
outputs=[gr.Image(type="pil", label="Image Output"), gr.Textbox(label="Prediction")],
|
33 |
title="Animals Classifier",
|
34 |
description="Upload an image of an animal, and the model will predict it.\n\n**Disclaimer:** This model is trained only on specific animal classes (butterfly, cats, cow, dogs, elephant, horse, monkey, sheep, spider, squirrel) and may not accurately predict animals outside these classes."
|
35 |
)
|