FoodBaseBERT
Model description
FoodBaseBERT is a fine-tuned BERT model that is ready to use for Named Entity Recognition of Food entities. It has been trained to recognize one entity: food (FOOD).
Specifically, this model is a bert-base-cased model that was fine-tuned on the FoodBase NER dataset.
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/FoodBaseBERT")
model = AutoModelForTokenClassification.from_pretrained("Dizex/FoodBaseBERT")
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)
- Downloads last month
- 354
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social
visibility and check back later, or deploy to Inference Endpoints (dedicated)
instead.