ChatFilter / download_models.py
John Doe
fix error
3ab5798
raw
history blame
196 Bytes
from transformers import pipeline
# This will download and cache the model
print("Downloading toxic-bert model...")
_ = pipeline("text-classification", model="unitary/toxic-bert")
print("Done.")