Spaces:
Runtime error
Runtime error
bonosa commited on
Commit Β·
26e94d3
1
Parent(s): 814099f
yay!
Browse files- app.py +1 -1
- 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('
|
| 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
|