LongAnam commited on
Commit
7f105d4
·
1 Parent(s): d878736

Change path of model

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -11,7 +11,8 @@ import gradio as gr
11
  def is_cat(x): return x[0].isupper()
12
 
13
  # %% ../app.ipynb 4
14
- learner = load_learner('model.pkl')
 
15
 
16
  # %% ../app.ipynb 6
17
  categories = ('Dog', 'Cat')
 
11
  def is_cat(x): return x[0].isupper()
12
 
13
  # %% ../app.ipynb 4
14
+ #learner = load_learner('model.pkl')
15
+ learner = from_pretrained_fastai("fastai/cat_or_dog")
16
 
17
  # %% ../app.ipynb 6
18
  categories = ('Dog', 'Cat')