Downloads · 30 days
29
1% of all-time downloads
flaubert-base-uncased-finetuned-cooking
nbouali/flaubert-base-uncased-finetuned-cooking
flaubert-base-uncased-finetuned-cooking is a text classification model from nbouali. Use it when you need a label for a piece of text. It is set up for transformers.
This model is finetuned on a sequence classification task that associates each sequence with the appropriate recipe category.
All-time downloads
2.2K
Public
Repo size
1.1 GB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.bin549 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- flaubert
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Apr 28, 2021
- SHA
- c21d936f
Try a prompt
Model card
From the Hugging Face model README
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
- Task
- Text Classification
- Library
- transformers
- Type
- flaubert
- Languages
- fr
- Created
- Mar 2, 2022
- Updated
- Apr 28, 2021