Kornel Szabo commited on
Commit
181d8f8
·
1 Parent(s): edb34ef

4th Commit

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -4,6 +4,7 @@
4
  __all__ = ['repo_id', 'learn', 'path', 'categories', 'image', 'output', 'examples', 'intf', 'second', 'classify_image']
5
 
6
  # %% 02_spider_deploy.ipynb 1
 
7
  from huggingface_hub import from_pretrained_fastai
8
  import gradio as gr
9
 
@@ -12,6 +13,7 @@ repo_id = "xt0r3/spider-classifier"
12
  learn = from_pretrained_fastai(repo_id)
13
 
14
  # %% 02_spider_deploy.ipynb 3
 
15
  categories = learn.dls.vocab
16
 
17
  # %% 02_spider_deploy.ipynb 4
 
4
  __all__ = ['repo_id', 'learn', 'path', 'categories', 'image', 'output', 'examples', 'intf', 'second', 'classify_image']
5
 
6
  # %% 02_spider_deploy.ipynb 1
7
+ from fastai.vision.all import *
8
  from huggingface_hub import from_pretrained_fastai
9
  import gradio as gr
10
 
 
13
  learn = from_pretrained_fastai(repo_id)
14
 
15
  # %% 02_spider_deploy.ipynb 3
16
+ path = Path("data/spiders")
17
  categories = learn.dls.vocab
18
 
19
  # %% 02_spider_deploy.ipynb 4