# TEST MODEL | |
from transformers import pipeline | |
repository_id="mskov/roberta-base-toxicity" | |
classifier = pipeline('text-classification',repository_id) | |
text = "Kederis proclaims innocence Olympic champion Kostas Kederis today left hospital ahead of his date with IOC inquisitors claiming his innocence and vowing: quot;After the crucifixion comes the resurrection. quot; .." | |
result = classifier(text) | |
predicted_label = result[0]["label"] | |
print(f"Predicted label: {predicted_label}") |