Spaces:
Runtime error
Runtime error
Salman Naqvi
commited on
Commit
·
b3324d1
1
Parent(s):
c2763a7
Switched the categories order so correct labels are output.
Browse files
app.ipynb
CHANGED
|
The diff for this file is too large to render.
See raw diff
|
|
|
app.py
CHANGED
|
@@ -11,7 +11,7 @@ import gradio as gr
|
|
| 11 |
learner = load_learner('model/export.pkl')
|
| 12 |
|
| 13 |
# %% app.ipynb 9
|
| 14 |
-
cats = ('
|
| 15 |
|
| 16 |
def classify_img(img):
|
| 17 |
preds, idx, probs = learner.predict(img)
|
|
|
|
| 11 |
learner = load_learner('model/export.pkl')
|
| 12 |
|
| 13 |
# %% app.ipynb 9
|
| 14 |
+
cats = ('Black Bear', 'Grizzly Bear', 'Teddy Bear',)
|
| 15 |
|
| 16 |
def classify_img(img):
|
| 17 |
preds, idx, probs = learner.predict(img)
|