Commit
•
e729418
1
Parent(s):
de6084e
Update app.py
Browse files
app.py
CHANGED
|
@@ -29,6 +29,8 @@ import gradio as gr
|
|
| 29 |
# %% ../fastai_lesson_2_bearsInference_colab.ipynb 12
|
| 30 |
learn = load_learner('bearClassifier.pkl')
|
| 31 |
|
|
|
|
|
|
|
| 32 |
# %% ../fastai_lesson_2_bearsInference_colab.ipynb 15
|
| 33 |
def classify_image(img):
|
| 34 |
pred,idx,probs = learn.predict(img)
|
|
|
|
| 29 |
# %% ../fastai_lesson_2_bearsInference_colab.ipynb 12
|
| 30 |
learn = load_learner('bearClassifier.pkl')
|
| 31 |
|
| 32 |
+
categories = ('grizzly', 'black', 'teddy')
|
| 33 |
+
|
| 34 |
# %% ../fastai_lesson_2_bearsInference_colab.ipynb 15
|
| 35 |
def classify_image(img):
|
| 36 |
pred,idx,probs = learn.predict(img)
|