reshane commited on
Commit
81d7092
·
verified ·
1 Parent(s): f2f571a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,12 +3,13 @@ __all__ = ["label_func", "learner", "classify_image", "categories", "image", "la
3
 
4
  from fastai.vision.all import *
5
  import gradio as gr
 
6
 
7
  def label_func(f): return f[0] == 'p'
8
  def acc_camvid(*_): pass
9
  def get_y(*_): pass
10
 
11
- learner = load_learner('my_export.pkl')
12
 
13
  categories = ('Not Poison Ivy', 'Poison Ivy')
14
 
 
3
 
4
  from fastai.vision.all import *
5
  import gradio as gr
6
+ from huggingface_hub import from_pretrained_fastai
7
 
8
  def label_func(f): return f[0] == 'p'
9
  def acc_camvid(*_): pass
10
  def get_y(*_): pass
11
 
12
+ learner = from_pretrained_fastai("reshane/PoisonIvy")
13
 
14
  categories = ('Not Poison Ivy', 'Poison Ivy')
15