Englios commited on
Commit
784b914
·
1 Parent(s): 2b9a1d1

Fix app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -5
app.py CHANGED
@@ -3,15 +3,18 @@ __all__ = ['getx','gety','learn','categories','classify_images','image','label',
3
  import gradio as gr
4
  from fastai.vision.all import *
5
 
6
- import pathlib
7
- plt = platform.system()
8
- if plt == 'Windows': pathlib.PosixPath = pathlib.WindowsPath
9
-
10
  def getx(r):return r['fname']
11
  def gety(r):return r['cat'].split(' ')
12
 
 
 
13
  model_path = Path('./model.pkl')
14
- learn = load_learner(model_path)
 
 
 
 
 
15
 
16
  categories = ('Grizzly','Black','Toy','White','Brown')
17
 
 
3
  import gradio as gr
4
  from fastai.vision.all import *
5
 
 
 
 
 
6
  def getx(r):return r['fname']
7
  def gety(r):return r['cat'].split(' ')
8
 
9
+ import pathlib
10
+ posix_backup = pathlib.PosixPath
11
  model_path = Path('./model.pkl')
12
+ try:
13
+ pathlib.PosixPath = pathlib.WindowsPath
14
+ learn = load_learner(model_path)
15
+ finally:
16
+ pathlib.PosixPath = posix_backup
17
+
18
 
19
  categories = ('Grizzly','Black','Toy','White','Brown')
20