Update app.py

app.py

@@ -13,7 +13,7 @@ def ner(text):
     doc = nlp(text)
     text = " ".join([token.text for token in doc])
     predictions = pl(text)
-    mentions = [pred["word"].strip() for pred in predictions if pred["entity_group"] == "LABEL_1"]   
+    mentions = [pred["word"].strip() for pred in predictions if pred["entity_group"] == "FOOD"]   
     return "\n".join(mentions)
 
 iface = gr.Interface(