zzarif commited on
Commit
615788b
·
1 Parent(s): b7fb918

windows path set to posix path

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -1,9 +1,11 @@
1
  from fastai.vision.all import *
2
  import gradio as gr
3
 
 
4
  import pathlib
5
- temp = pathlib.PosixPath
6
- pathlib.PosixPath = pathlib.WindowsPath
 
7
 
8
  pasta_shape_labels = (
9
  "bucatini",
@@ -28,7 +30,7 @@ pasta_shape_labels = (
28
  "tortellini"
29
  )
30
 
31
- model = load_learner(pathlib.Path('pasta_shape_recognizer_v2.pkl'))
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)