Harald commited on
Commit
64d956c
·
1 Parent(s): aae67a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -1,9 +1,11 @@
1
  import gradio as gr
2
  from fastai.vision.all import *
3
- from pathlib import Path
4
 
5
- model_path = Path('export.pkl')
6
- learn = load_learner(model_path)
 
 
7
  categories = ('Black', 'Grizzly', 'Teddy')
8
 
9
  def classify_image(img):
 
1
  import gradio as gr
2
  from fastai.vision.all import *
3
+ import pathlib
4
 
5
+ plt = platform.system()
6
+ if plt == 'Windows': pathlib.WindowsPath = pathlib.PosixPath
7
+
8
+ learn = load_learner('export.pkl')
9
  categories = ('Black', 'Grizzly', 'Teddy')
10
 
11
  def classify_image(img):