Spaces:
Sleeping
Sleeping
Asis
commited on
Commit
•
067eb84
1
Parent(s):
ed9f87c
Rename labels
Browse files
app.py
CHANGED
@@ -6,8 +6,8 @@ learn = load_learner('model.pkl')
|
|
6 |
|
7 |
labels = learn.dls.vocab
|
8 |
|
9 |
-
labels = list(map(lambda x: x.replace('food recipe', '
|
10 |
-
labels = list(map(lambda x: x.replace('empty plate', 'Empty
|
11 |
|
12 |
def predict(img):
|
13 |
img = PILImage.create(img)
|
|
|
6 |
|
7 |
labels = learn.dls.vocab
|
8 |
|
9 |
+
labels = list(map(lambda x: x.replace('food recipe', 'Plate with food'), labels))
|
10 |
+
labels = list(map(lambda x: x.replace('empty plate', 'Empty plate'), labels))
|
11 |
|
12 |
def predict(img):
|
13 |
img = PILImage.create(img)
|