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')
|