socd06 commited on
Commit
33bfc8e
·
1 Parent(s): 384db06

upload pickled model and call it during inference

Browse files
Files changed (2) hide show
  1. app.py +2 -4
  2. model.pkl +3 -0
app.py CHANGED
@@ -1,13 +1,11 @@
1
  import os
2
  from huggingface_hub import from_pretrained_fastai
3
-
4
  import gradio as gr
5
  from gradio.components import Image
6
  from gradio import Label
7
 
8
- SECRET_TOKEN = os.getenv("SECRET_TOKEN")
9
- repo_id = "artificeresearch/spiritvision"
10
- learner = from_pretrained_fastai(repo_id)
11
  labels = learner.dls.vocab
12
 
13
 
 
1
  import os
2
  from huggingface_hub import from_pretrained_fastai
3
+ from fastai.vision.all import *
4
  import gradio as gr
5
  from gradio.components import Image
6
  from gradio import Label
7
 
8
+ learner = load_learner('model.pkl')
 
 
9
  labels = learner.dls.vocab
10
 
11
 
model.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c7424eb5cd09cf7d32c11035cfa43fb56d242e250445048d134e61b3645d6ffd
3
+ size 102878171