Spaces:
Runtime error
Runtime error
Commit ·
1d985f4
1
Parent(s): b08d721
Added example images. Moved categories before function.
Browse files- app.py +1 -3
- bear1.jpeg +0 -0
- bear2.jpeg +0 -0
- bear3.jpeg +0 -0
app.py
CHANGED
|
@@ -7,14 +7,12 @@ warnings.filterwarnings("ignore", category=UserWarning, module="fastai.learner")
|
|
| 7 |
|
| 8 |
learn = load_learner('export.pkl')
|
| 9 |
|
|
|
|
| 10 |
|
| 11 |
def classify_bear(image):
|
| 12 |
pred, pred_idx, probs = learn.predict(image)
|
| 13 |
return dict(zip(categories, map(float, probs)))
|
| 14 |
|
| 15 |
-
|
| 16 |
-
categories = ('Black', 'Grizzly', 'Teddy')
|
| 17 |
-
|
| 18 |
image = gr.Image(height=192, width=192)
|
| 19 |
label = gr.Label(num_top_classes=3)
|
| 20 |
examples = ['bear1.jpeg', 'bear2.jpeg', 'bear3.jpeg']
|
|
|
|
| 7 |
|
| 8 |
learn = load_learner('export.pkl')
|
| 9 |
|
| 10 |
+
categories = ('Black', 'Grizzly', 'Teddy')
|
| 11 |
|
| 12 |
def classify_bear(image):
|
| 13 |
pred, pred_idx, probs = learn.predict(image)
|
| 14 |
return dict(zip(categories, map(float, probs)))
|
| 15 |
|
|
|
|
|
|
|
|
|
|
| 16 |
image = gr.Image(height=192, width=192)
|
| 17 |
label = gr.Label(num_top_classes=3)
|
| 18 |
examples = ['bear1.jpeg', 'bear2.jpeg', 'bear3.jpeg']
|
bear1.jpeg
ADDED
|
bear2.jpeg
ADDED
|
bear3.jpeg
ADDED
|