practic / app.py
SimrusDenuvo's picture
Update app.py
22193d7 verified
raw
history blame contribute delete
544 Bytes
from transformers import pipeline
clf = pipeline(
task = 'sentiment-analysis',
model = 'SkolkovoInstitute/russian_toxicity_classifier')
text = ['У нас в есть убунты и текникал превью.',
'Как минимум два малолетних дегенерата в треде, мда.']
def data(text):
for row in text:
yield row
for out in clf(data(text)):
print(out)
#вывод
{'label': 'neutral', 'score': 0.9872767329216003}
{'label': 'toxic', 'score': 0.985331654548645}