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
15
Hosted inference API
Text Classification
Examples
Examples
This model can be loaded on the Inference API on-demand.