Update app.py
Browse files
app.py
CHANGED
|
@@ -79,7 +79,6 @@ def load_models_and_tokenizers(hf_token):
|
|
| 79 |
|
| 80 |
# Load other models
|
| 81 |
model_configs = [
|
| 82 |
-
("unitary/toxic-bert", AutoModelForSequenceClassification, "unitary/toxic-bert"),
|
| 83 |
("s-nlp/roberta_toxicity_classifier", AutoModelForSequenceClassification, "s-nlp/roberta_toxicity_classifier"),
|
| 84 |
("martin-ha/toxic-comment-model", AutoModelForSequenceClassification, "martin-ha/toxic-comment-model"),
|
| 85 |
("lmsys/toxicchat-t5-large-v1.0", AutoModelForSeq2SeqLM, "t5-large")
|
|
|
|
| 79 |
|
| 80 |
# Load other models
|
| 81 |
model_configs = [
|
|
|
|
| 82 |
("s-nlp/roberta_toxicity_classifier", AutoModelForSequenceClassification, "s-nlp/roberta_toxicity_classifier"),
|
| 83 |
("martin-ha/toxic-comment-model", AutoModelForSequenceClassification, "martin-ha/toxic-comment-model"),
|
| 84 |
("lmsys/toxicchat-t5-large-v1.0", AutoModelForSeq2SeqLM, "t5-large")
|