Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2from transformers import TextClassificationPipeline
3
4loaded_tokenizer = AutoTokenizer.from_pretrained("nbouali/flaubert-base-uncased-finetuned-cooking")
5loaded_model = AutoModelForSequenceClassification.from_pretrained("nbouali/flaubert-base-uncased-finetuned-cooking")
6
7nlp = TextClassificationPipeline(model=loaded_model,tokenizer=loaded_tokenizer,task="Recipe classification")
8
9print(nlp("Lasagnes à la bolognaise"))[{'label': 'LABEL_6', 'score': 0.9921900033950806}]| 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