Views
No views yet
1from transformers import pipeline
2
3pipe = pipeline("text-classification",
4 model = "camilum/emotions-DistilBETO",
5 tokenizer = "dccuchile/distilbert-base-spanish-uncased", # NOTE Do not delete this
6 padding = True,
7 truncation = "longest_first",
8 top_k = None,
9)
10
11print(list(pipe("estoy triste")))
12print(list(pipe("estoy triste " * 1024)))[[{'label': 'LABEL_6', 'score': 0.9897441267967224}, {'label': 'LABEL_0', 'score': 0.004266336094588041}, {'label': 'LABEL_2', 'score': 0.0014442066894844174}, {'label': 'LABEL_1', 'score': 0.001328755053691566}, {'label': 'LABEL_5', 'score': 0.001040525734424591}, {'label': 'LABEL_3', 'score': 0.0009753472404554486}, {'label': 'LABEL_4', 'score': 0.0008171153021976352}, {'label': 'LABEL_7', 'score': 0.0003837039403151721}]]
[[{'label': 'LABEL_6', 'score': 0.9416162967681885}, {'label': 'LABEL_0', 'score': 0.03937496617436409}, {'label': 'LABEL_1', 'score': 0.00783670973032713}, {'label': 'LABEL_2', 'score': 0.004338116850703955}, {'label': 'LABEL_5', 'score': 0.0027957989368587732}, {'label': 'LABEL_4', 'score': 0.0017090848414227366}, {'label': 'LABEL_3', 'score': 0.0014251844258978963}, {'label': 'LABEL_7', 'score': 0.0009038725402206182}]]transformers==4.50.3
torch==2.6.0