javierosello commited on
Commit
f2b7b42
·
verified ·
1 Parent(s): 3226d37

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -14,6 +14,7 @@ from huggingface_hub import from_pretrained_fastai, notebook_login, push_to_hub_
14
  import timm
15
  from torchvision.models import resnet18
16
  import os
 
17
 
18
  learn = load_learner('model.pkl')
19
 
 
14
  import timm
15
  from torchvision.models import resnet18
16
  import os
17
+ from fastapi.learner import load_learner
18
 
19
  learn = load_learner('model.pkl')
20