jeffreymjohnson commited on
Commit
92b42f4
·
1 Parent(s): ced6706

changed library to huggingface

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
- from fastai.vision.all import *
2
  import gradio as gr
3
 
4
- learn = load_learner('./deadpool-detection-model.pkl')
5
 
6
  labels = learn.dls.vocab
7
 
 
1
+ from huggingface_hub import from_pretrained_fastai
2
  import gradio as gr
3
 
4
+ learn = from_pretrained_fastai('./deadpool-detection-model.pkl')
5
 
6
  labels = learn.dls.vocab
7