Yichuan Huang commited on
Commit
3a51bc4
·
verified ·
1 Parent(s): bcfd061

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -37,8 +37,14 @@ interface = gr.Interface(
37
  fn=predict,
38
  inputs=gr.Image(type="pil"),
39
  outputs=gr.Label(num_top_classes=5),
40
- examples=[["input_imgs/cat.jpeg"], ["input_imgs/dog.jpeg"]],
41
- title="Animal Classifier",
 
 
 
 
 
 
42
  description="This is a demo of a resnet50 model trained on COCO dataset, which can classify 5 classes: bird, cat, dog, horse, sheep.",
43
  )
44
 
 
37
  fn=predict,
38
  inputs=gr.Image(type="pil"),
39
  outputs=gr.Label(num_top_classes=5),
40
+ examples=[
41
+ ["input_imgs/bird.jpeg"],
42
+ ["input_imgs/cat.jpeg"],
43
+ ["input_imgs/dog.jpeg"],
44
+ ["input_imgs/horse.jpeg"],
45
+ ["input_imgs/sheep.jpeg"],
46
+ ],
47
+ title="Image Object Classifier",
48
  description="This is a demo of a resnet50 model trained on COCO dataset, which can classify 5 classes: bird, cat, dog, horse, sheep.",
49
  )
50