Views
No views yet
pipelines 🚀1# pip install -q transformers
2
3from transformers import AutoModelForSequenceClassification, AutoTokenizer
4
5CHKPT = 'mrm8488/electricidad-base-finetuned-muchocine'
6model = AutoModelForSequenceClassification.from_pretrained(CHKPT)
7tokenizer = AutoTokenizer.from_pretrained(CHKPT)
8
9from transformers import pipeline
10classifier = pipeline('sentiment-analysis', model=model, tokenizer=tokenizer)
11
12# It ranks your comments between 1 and 5 (stars)
13
14classifier('Es una obra mestra. Brillante.')
15# [{'label': '5', 'score': 0.9498381614685059}]
16
17classifier('Es una película muy buena.')
18# {'label': '4', 'score': 0.9277070760726929}]
19
20classifier('Una buena película, sin más.')
21# [{'label': '3', 'score': 0.9768431782722473}]
22
23classifier('Esperaba mucho más.')
24# [{'label': '2', 'score': 0.7063605189323425}]
25
26classifier('He tirado el dinero. Una basura. Vergonzoso.')
27# [{'label': '1', 'score': 0.8494752049446106}]