Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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('
|
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 |
-
'
|
40 |
-
'
|
41 |
-
'
|
42 |
-
'
|
43 |
-
'
|
44 |
-
'
|
45 |
-
'
|
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)
|