Spaces:
Sleeping
Sleeping
Added Categories
Browse files
app.py
CHANGED
|
@@ -10,6 +10,9 @@ def classify_digit_type(img):
|
|
| 10 |
|
| 11 |
learn = load_learner('model.pkl')
|
| 12 |
|
|
|
|
|
|
|
|
|
|
| 13 |
labels = learn.dls.vocab
|
| 14 |
def predict(img):
|
| 15 |
img = PILImage.create(img)
|
|
|
|
| 10 |
|
| 11 |
learn = load_learner('model.pkl')
|
| 12 |
|
| 13 |
+
categories = ('0', '1', '2', '3', '4', '5', '6', '7', '8', '9')
|
| 14 |
+
|
| 15 |
+
|
| 16 |
labels = learn.dls.vocab
|
| 17 |
def predict(img):
|
| 18 |
img = PILImage.create(img)
|