Spaces:
Runtime error
Runtime error
Commit
·
0266199
1
Parent(s):
406195b
updated settings
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ import skimage
|
|
| 6 |
|
| 7 |
learner = load_learner('model.pkl')
|
| 8 |
|
| 9 |
-
labels =
|
| 10 |
def classify_image(img):
|
| 11 |
pred, idx, probs = learner.predict(img)
|
| 12 |
return dict(zip(learner.dls.vocab, map(float, probs)))
|
|
|
|
| 6 |
|
| 7 |
learner = load_learner('model.pkl')
|
| 8 |
|
| 9 |
+
labels = learner.dls.vocab
|
| 10 |
def classify_image(img):
|
| 11 |
pred, idx, probs = learner.predict(img)
|
| 12 |
return dict(zip(learner.dls.vocab, map(float, probs)))
|