Schib commited on
Commit
e593fd6
1 Parent(s): b815718

fix paths and add images

Browse files
Files changed (4) hide show
  1. app.py +2 -2
  2. kitchen.jpeg +0 -0
  3. not_kitchen.jpeg +0 -0
  4. outside.jpeg +0 -0
app.py CHANGED
@@ -3,7 +3,7 @@ from fastai.vision.all import *
3
  import gradio as gr
4
 
5
  # Cell
6
- learn = load_learner('/kaggle/input/kitchen-detector-model/model.pkl')
7
 
8
  # Cell
9
  categories = ('kitchen', 'not_kitchen')
@@ -15,7 +15,7 @@ def classify_image(img):
15
  # Cell
16
  image = gr.inputs.Image(shape=(192, 192))
17
  label = gr.outputs.Label()
18
- examples = ['/kaggle/input/kitchen-data-set/kitchen-data-set/kitchen/209154559.jpeg', '/kaggle/input/kitchen-data-set/kitchen-data-set/not_kitchen/209154922.jpeg']
19
 
20
  intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
21
  intf.launch(inline=False)
 
3
  import gradio as gr
4
 
5
  # Cell
6
+ learn = load_learner('model.pkl')
7
 
8
  # Cell
9
  categories = ('kitchen', 'not_kitchen')
 
15
  # Cell
16
  image = gr.inputs.Image(shape=(192, 192))
17
  label = gr.outputs.Label()
18
+ examples = ['kitchen.jpeg', 'not_kitchen.jpeg', 'outside.jpeg']
19
 
20
  intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
21
  intf.launch(inline=False)
kitchen.jpeg ADDED
not_kitchen.jpeg ADDED
outside.jpeg ADDED