Spaces:
Runtime error
Runtime error
Commit Β·
a896a7a
1
Parent(s): 789964f
categories
Browse files
app.py
CHANGED
|
@@ -5,7 +5,7 @@ from fastai.vision.all import *
|
|
| 5 |
learn = load_learner('model.pkl')
|
| 6 |
|
| 7 |
examples = ['apples.webp', 'avocadoes.jpg', 'bananas.jpg', 'blueberries.jpg', 'grapes.jpg', 'mangos.jpg', 'oranges.jpg', 'pineapples.jpg', 'strawberries.jpg', 'watermelon.jpg']
|
| 8 |
-
categories = ('
|
| 9 |
def classify_image(img):
|
| 10 |
pred, idx, probs = learn.predict(img)
|
| 11 |
return dict(zip(categories, map(float, probs)))
|
|
|
|
| 5 |
learn = load_learner('model.pkl')
|
| 6 |
|
| 7 |
examples = ['apples.webp', 'avocadoes.jpg', 'bananas.jpg', 'blueberries.jpg', 'grapes.jpg', 'mangos.jpg', 'oranges.jpg', 'pineapples.jpg', 'strawberries.jpg', 'watermelon.jpg']
|
| 8 |
+
categories = ('apples', 'avocados', 'bananas', 'bluberries', 'grapes', 'mangoes', 'oranges', 'pineapples', 'strawberries', 'watermelon')
|
| 9 |
def classify_image(img):
|
| 10 |
pred, idx, probs = learn.predict(img)
|
| 11 |
return dict(zip(categories, map(float, probs)))
|