jonathanIckovich commited on
Commit
3f19ca5
·
1 Parent(s): d0acbbb
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,7 +1,8 @@
1
  import gradio as gr
2
  from fastai.vision.all import *
3
 
4
- learn = load_learner('ImageDifferentiator.pkl')
 
5
  labels = learn.dls.vocab
6
  def predict(img):
7
 
 
1
  import gradio as gr
2
  from fastai.vision.all import *
3
 
4
+ model_path = Path('ImageDifferentiator.pkl')
5
+ learn = load_learner(model_path)
6
  labels = learn.dls.vocab
7
  def predict(img):
8