Edit labels

app.py

@@ -5,6 +5,10 @@ import skimage
 learn = load_learner('model.pkl')
 
 labels = learn.dls.vocab
+
+labels = list(map(lambda x: x.replace('food recipe', 'Food Plate'), labels))
+labels = list(map(lambda x: x.replace('empty plate', 'Empty Plate'), labels))
+
 def predict(img):
     img = PILImage.create(img)
     pred,pred_idx,probs = learn.predict(img)
@@ -18,3 +22,4 @@ interpretation = 'default'
 enable_queue = True
 
 gr.Interface(fn=predict,inputs=gr.inputs.Image(shape=(512, 512)),outputs=gr.outputs.Label(num_top_classes=3),title=title,description=description,article=article,examples=examples,interpretation=interpretation,enable_queue=enable_queue).launch()
+