Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2
3repo = "clairedhx/camembert_dp_ft_top40_bdx"
4tok = AutoTokenizer.from_pretrained(repo, use_fast=True)
5model = AutoModelForSequenceClassification.from_pretrained(repo)
6
7x = tok("Douleur thoracique avec sus-décalage ST...", return_tensors="pt")
8pred_id = int(model(**x).logits.argmax(-1))
9print(model.config.id2label[pred_id])