bonosa commited on
Commit
26e94d3
Β·
1 Parent(s): 814099f
Files changed (2) hide show
  1. app.py +1 -1
  2. parrotclasscolab.pkl +3 -0
app.py CHANGED
@@ -12,7 +12,7 @@ def custom_resnet_splitter(model):
12
  return [params(resnet[0]), params(resnet[1]), params(resnet[4]), params(resnet[5]), params(resnet[6]), params(resnet[7]), params(model[1])]
13
 
14
  def predict_parrot_species(image):
15
- model_path = os.path.abspath('parrotclass.pkl') # Resolve the path object
16
  learn_inf = load_learner(str(model_path)) # Convert the Path object to a string
17
  pred, _, _ = learn_inf.predict(image)
18
  return pred
 
12
  return [params(resnet[0]), params(resnet[1]), params(resnet[4]), params(resnet[5]), params(resnet[6]), params(resnet[7]), params(model[1])]
13
 
14
  def predict_parrot_species(image):
15
+ model_path = os.path.abspath('parrotclasscolab.pkl') # Resolve the path object
16
  learn_inf = load_learner(str(model_path)) # Convert the Path object to a string
17
  pred, _, _ = learn_inf.predict(image)
18
  return pred
parrotclasscolab.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:744685096120a1d9789ebbb800645d68e05ea78bf4e06fa024ccac3ef90a540a
3
+ size 46974479