Alex Strick van Linschoten commited on
Commit
b656c5a
·
1 Parent(s): f84e090

upgrade using huggingface hub

Browse files
Files changed (2) hide show
  1. app.py +6 -1
  2. model.pkl +0 -3
app.py CHANGED
@@ -2,7 +2,12 @@ import gradio as gr
2
  from fastai.vision.all import *
3
  import skimage
4
 
5
- learn = load_learner("model.pkl")
 
 
 
 
 
6
 
7
  labels = learn.dls.vocab
8
 
 
2
  from fastai.vision.all import *
3
  import skimage
4
 
5
+ from huggingface_hub import hf_hub_download
6
+ from fastai.learner import load_learner
7
+
8
+ learn = load_learner(
9
+ hf_hub_download("strickvl/redaction-classifier-fastai", "model.pkl")
10
+ )
11
 
12
  labels = learn.dls.vocab
13
 
model.pkl DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:f1f259b0d96a72e65b72fea92226109aebf275adeec7e1dcddda00e8cb5b8ec7
3
- size 87538219