shinmemkuf commited on
Commit
22a39bc
·
1 Parent(s): bb37043

changed learner

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -1,9 +1,12 @@
1
  from fastai.vision.all import *
 
2
  import gradio as gr
3
 
4
  def is_cat(x): return x[0].isupper()
5
 
6
- learn = load_learner('model.pkl')
 
 
7
 
8
  categories = ('Dog', 'Cat')
9
 
 
1
  from fastai.vision.all import *
2
+ from huggingface_hub import from_pretrained_fastai
3
  import gradio as gr
4
 
5
  def is_cat(x): return x[0].isupper()
6
 
7
+ # repo_id = "YOUR_USERNAME/YOUR_LEARNER_NAME"
8
+ repo_id = "shinmemkuf/lesson2_demo"
9
+ learn = from_pretrained_fastai(repo_id)
10
 
11
  categories = ('Dog', 'Cat')
12