dielz commited on
Commit
0de90f8
·
verified ·
1 Parent(s): 1a94697

Fix labels error

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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="Predictions"],
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
  )