Spaces:
Runtime error
Runtime error
windows path set to posix path
Browse files
app.py
CHANGED
@@ -1,9 +1,11 @@
|
|
1 |
from fastai.vision.all import *
|
2 |
import gradio as gr
|
3 |
|
|
|
4 |
import pathlib
|
5 |
-
|
6 |
-
|
|
|
7 |
|
8 |
pasta_shape_labels = (
|
9 |
"bucatini",
|
@@ -28,7 +30,7 @@ pasta_shape_labels = (
|
|
28 |
"tortellini"
|
29 |
)
|
30 |
|
31 |
-
model = load_learner(
|
32 |
|
33 |
def recognize_image(image):
|
34 |
pred, idx, probs = model.predict(image)
|
|
|
1 |
from fastai.vision.all import *
|
2 |
import gradio as gr
|
3 |
|
4 |
+
import platform
|
5 |
import pathlib
|
6 |
+
plt = platform.system()
|
7 |
+
if plt == 'Linux':
|
8 |
+
pathlib.WindowsPath = pathlib.PosixPath
|
9 |
|
10 |
pasta_shape_labels = (
|
11 |
"bucatini",
|
|
|
30 |
"tortellini"
|
31 |
)
|
32 |
|
33 |
+
model = load_learner('pasta_shape_recognizer_v2.pkl')
|
34 |
|
35 |
def recognize_image(image):
|
36 |
pred, idx, probs = model.predict(image)
|