Second commit

app.py

@@ -1,22 +1,22 @@
 # AUTOGENERATED! DO NOT EDIT! File to edit: 02_spider_deploy.ipynb.
 
 # %% auto 0
-__all__ = ['learn', 'path', 'categories', 'image', 'output', 'examples', 'intf', 'second', 'classify_image']
+__all__ = ['repo_id', 'learn', 'path', 'categories', 'image', 'output', 'examples', 'intf', 'second', 'classify_image']
 
-# %% 02_spider_deploy.ipynb 0
-from fastbook import *
+# %% 02_spider_deploy.ipynb 1
+from fastai.vision.all import *
+from huggingface_hub import from_pretrained_fastai
+import gradio as gr
 
 # %% 02_spider_deploy.ipynb 2
-import gradio as gr
+repo_id = "xt0r3/spider-classifier"
+learn = from_pretrained_fastai(repo_id)
 
 # %% 02_spider_deploy.ipynb 3
-learn = load_learner("spider.pkl")
-
-# %% 02_spider_deploy.ipynb 4
 path = Path("data/spiders")
 categories = learn.dls.vocab
 
-# %% 02_spider_deploy.ipynb 5
+# %% 02_spider_deploy.ipynb 4
 def second(x):
     a, b = x
     return b
@@ -30,7 +30,7 @@ def classify_image(img):
     results = sorted(results, key=second, reverse=True)
     return dict(results[:10])
 
-# %% 02_spider_deploy.ipynb 7
+# %% 02_spider_deploy.ipynb 6
 image = gr.inputs.Image()
 output = gr.outputs.Label()
 examples = ["data/spider1.jpg", "data/spider2.jpg"]