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
3
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.