Spaces:
Sleeping
Sleeping
socd06
commited on
Commit
·
33bfc8e
1
Parent(s):
384db06
upload pickled model and call it during inference
Browse files
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 |
-
|
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
|