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