Regev89 commited on
Commit
d92ca28
1 Parent(s): 9f78a58
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -7,10 +7,10 @@ def classify_img(img):
7
 
8
  image = gr.Image()
9
  label = gr.Label()
10
- examples = ['examples/grizzly.jpg', 'examples/black.jpg', 'examples/teddy.jpg']
11
 
12
- model = load_learner('bears.pkl')
13
 
14
- intf = gr.Interface(fn=lambda image: "grizzly bear", inputs=image, outputs=label, examples=examples)
15
  iface = gr.Interface(fn=classify_img, inputs=image, outputs=label, examples=examples)
16
  iface.launch()
 
7
 
8
  image = gr.Image()
9
  label = gr.Label()
10
+ examples = ['examples/pizza.jpg', 'examples/burger.jpg', 'examples/ice cream.jpg']
11
 
12
+ model = load_learner('foods.pkl')
13
 
14
+ intf = gr.Interface(fn=lambda image: "pizza", inputs=image, outputs=label, examples=examples)
15
  iface = gr.Interface(fn=classify_img, inputs=image, outputs=label, examples=examples)
16
  iface.launch()