Change path of model
Browse files
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')
|