Rhasan97 commited on
Commit
8f50e73
1 Parent(s): fdd0746

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -11
app.py CHANGED
@@ -1,9 +1,9 @@
1
  from fastai.vision.all import load_learner
2
  import gradio as gr
3
 
4
- import pathlib
5
- temp=pathlib.PosixPath
6
- pathlib.PosixPath=pathlib.WindowsPath
7
 
8
  snake_labels = (
9
  "Monocled cobra",
@@ -26,7 +26,7 @@ snake_labels = (
26
  "King cobra"
27
  )
28
 
29
- model = load_learner('models/snake-recognizer-v0.pkl')
30
 
31
  def recognize_snake(photo):
32
  pred, idx, probs = model.predict(photo)
@@ -36,13 +36,13 @@ def recognize_snake(photo):
36
  image = gr.inputs.Image(shape=(192,192))
37
  label = gr.outputs.Label()
38
  examples = [
39
- 'test data/viper-2.jpg',
40
- 'test data/shutterstock_2062214282-edited-1-scaled.jpg',
41
- 'test data/download (3).jpg',
42
- 'test data/download (4).jpg',
43
- 'test data/Naja_sputatrix.jpg',
44
- 'test data/download (6).jpg',
45
- 'test data/download.jpg'
46
  ]
47
 
48
  iface = gr.Interface(fn=recognize_snake, inputs=image, outputs=label, examples=examples)
 
1
  from fastai.vision.all import load_learner
2
  import gradio as gr
3
 
4
+ #import pathlib
5
+ #temp=pathlib.PosixPath
6
+ #pathlib.PosixPath=pathlib.WindowsPath
7
 
8
  snake_labels = (
9
  "Monocled cobra",
 
26
  "King cobra"
27
  )
28
 
29
+ model = load_learner('snake-recognizer-v0.pkl')
30
 
31
  def recognize_snake(photo):
32
  pred, idx, probs = model.predict(photo)
 
36
  image = gr.inputs.Image(shape=(192,192))
37
  label = gr.outputs.Label()
38
  examples = [
39
+ 'viper-2.jpg',
40
+ 'shutterstock_2062214282-edited-1-scaled.jpg',
41
+ 'download (3).jpg',
42
+ 'download (4).jpg',
43
+ 'Naja_sputatrix.jpg',
44
+ 'download (6).jpg',
45
+ 'download.jpg'
46
  ]
47
 
48
  iface = gr.Interface(fn=recognize_snake, inputs=image, outputs=label, examples=examples)