test / app.py
mskov's picture
Update app.py
7e8d389
raw
history blame
539 Bytes
# TEST MODEL
from transformers import pipeline
classifier = pipeline(task="zero-shot-audio-classification", model="mskov/roberta-base-toxicity")
classifier = pipeline(model="mskov/roberta-base-toxicity")
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}")