bonosa commited on
Commit
37d1ad6
Β·
1 Parent(s): c0a3e8f
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ def custom_resnet_splitter(model):
18
  return [params(resnet[0]), params(resnet[1]), params(resnet[4]), params(resnet[5]), params(resnet[6]), params(resnet[7]), params(model[1])]
19
 
20
  def predict_parrot_species(image):
21
- learn_inf = load_learner(Path('parrotclass.pkl'))
22
  pred, _, _ = learn_inf.predict(image)
23
  return pred
24
 
 
18
  return [params(resnet[0]), params(resnet[1]), params(resnet[4]), params(resnet[5]), params(resnet[6]), params(resnet[7]), params(model[1])]
19
 
20
  def predict_parrot_species(image):
21
+ learn_inf = load_learner(str(Path('parrotclass.pkl')))
22
  pred, _, _ = learn_inf.predict(image)
23
  return pred
24