Spaces:
Runtime error
Runtime error
bonosa commited on
Commit Β·
37d1ad6
1
Parent(s): c0a3e8f
yay!
Browse files
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 |
|