Asis commited on
Commit
067eb84
1 Parent(s): ed9f87c

Rename labels

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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', 'Food Plate'), labels))
10
- labels = list(map(lambda x: x.replace('empty plate', 'Empty Plate'), labels))
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)