Edit model card

FlauBERT finetuned on French cooking recipes

This model is finetuned on a sequence classification task that associates each sequence with the appropriate recipe category.

How to use it?

from transformers import AutoTokenizer, AutoModelForSequenceClassification
from transformers import TextClassificationPipeline

loaded_tokenizer = AutoTokenizer.from_pretrained("nbouali/flaubert-base-uncased-finetuned-cooking")
loaded_model = AutoModelForSequenceClassification.from_pretrained("nbouali/flaubert-base-uncased-finetuned-cooking")

nlp = TextClassificationPipeline(model=loaded_model,tokenizer=loaded_tokenizer,task="Recipe classification")

print(nlp("Lasagnes à la bolognaise"))
[{'label': 'LABEL_6', 'score': 0.9921900033950806}]

Label encoding:

label Recipe Category
0 'Accompagnement'
1 'Amuse-gueule'
2 'Boisson'
3 'Confiserie'
4 'Dessert'
5 'Entrée'
6 'Plat principal'
7 'Sauce'

If you would like to know more about this model you can refer to our blog post

Downloads last month
Hosted inference API
Text Classification
This model can be loaded on the Inference API on-demand.