Hugging Face's logo

Spaces:
devudilip
/
fastai_pet_classifier
Sleeping

App Files Community
devudilip commited on
Commit
5b5dc4f
·
verified ·
1 Parent(s): adfc461

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import dill
4
 
5
  def is_cat(x): return x[0].isupper()
6
 
7
- learn = load_learner('model.pkl')
8
 
9
 
10
  categories = ('Dog', 'Cat')
 
4
 
5
  def is_cat(x): return x[0].isupper()
6
 
7
+ learn = load_learner('model.pkl', pickle_module=dill)
8
 
9
 
10
  categories = ('Dog', 'Cat')