shahp7575 commited on
Commit
5e97f7b
1 Parent(s): 1be0b6e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1,7 +1,7 @@
1
  from fastai.vision.all import *
2
  import gradio as gr
3
 
4
- learn = load_learner("model/what_brew_machine_v1.pkl")
5
 
6
  categories = tuple(learn.dls.vocab)
7
 
@@ -11,9 +11,9 @@ def what_machine(img):
11
 
12
  image = gr.inputs.Image(shape=(192,192))
13
  label = gr.outputs.Label()
14
- examples = ['sample_imgs/aeropress_go.jpg',
15
- 'sample_imgs/delonghi_espresso.jpg',
16
- 'sample_imgs/moka_pot_red.jpeg']
17
 
18
  intf = gr.Interface(fn=what_machine, inputs=image, outputs=label, examples=examples)
19
  intf.launch(inline=False)
 
1
  from fastai.vision.all import *
2
  import gradio as gr
3
 
4
+ learn = load_learner("what_brew_machine_v1.pkl")
5
 
6
  categories = tuple(learn.dls.vocab)
7
 
 
11
 
12
  image = gr.inputs.Image(shape=(192,192))
13
  label = gr.outputs.Label()
14
+ examples = ['aeropress_go.jpg',
15
+ 'delonghi_espresso.jpg',
16
+ 'moka_pot_red.jpeg']
17
 
18
  intf = gr.Interface(fn=what_machine, inputs=image, outputs=label, examples=examples)
19
  intf.launch(inline=False)