InstaFoodBERT-NER / README.md
Dizex's picture
Update README.md
2aa3968
metadata
language: en
datasets:
  - Dizex/InstaFoodSet
widget:
  - text: >-
      Today's meal: Fresh olive poké bowl topped with chia seeds. Very
      delicious!
    example_title: Food example 1
  - text: >-
      Tartufo Pasta with garlic flavoured butter and olive oil, egg yolk,
      parmigiano and pasta water.
    example_title: Food example 2
tags:
  - Instagram
  - NER
  - Named Entity Recognition
  - Food Entity Extraction
  - Social Media
  - Informal text
license: mit

InstaFoodBERT-NER

Model description

InstaFoodBERT-NER is a fine-tuned BERT model that is ready to use for Named Entity Recognition of Food entities on informal text (social media like). It has been trained to recognize a single entity: food (FOOD).

Specifically, this model is a bert-base-cased model that was fine-tuned on a dataset consisting of 400 English Instagram posts related to food. The dataset is open source.

Intended uses

How to use

You can use this model with Transformers pipeline for NER.

from transformers import AutoTokenizer, AutoModelForTokenClassification
from transformers import pipeline

tokenizer = AutoTokenizer.from_pretrained("Dizex/InstaFoodBERT-NER")
model = AutoModelForTokenClassification.from_pretrained("Dizex/InstaFoodBERT-NER")

pipe = pipeline("ner", model=model, tokenizer=tokenizer)
example = "Today's meal: Fresh olive poké bowl topped with chia seeds. Very delicious!"

ner_entity_results = pipe(example)
print(ner_entity_results)