Update app.py
Browse files
app.py
CHANGED
|
@@ -9,7 +9,8 @@ import gradio as gr
|
|
| 9 |
def is_cat(x): return x[0].isupper()
|
| 10 |
|
| 11 |
# Cell
|
| 12 |
-
learn = load_learner('model.pkl')
|
|
|
|
| 13 |
|
| 14 |
# Cell
|
| 15 |
categories = ('Dog', 'Cat')
|
|
|
|
| 9 |
def is_cat(x): return x[0].isupper()
|
| 10 |
|
| 11 |
# Cell
|
| 12 |
+
# learn = load_learner('model.pkl')
|
| 13 |
+
learn = load_learner('resnet18_0.pkl')
|
| 14 |
|
| 15 |
# Cell
|
| 16 |
categories = ('Dog', 'Cat')
|