Балаганский Никита Николаевич
commited on
Commit
•
aaf61e5
1
Parent(s):
1e2066e
fix order
Browse files
app.py
CHANGED
|
@@ -23,8 +23,8 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
| 23 |
|
| 24 |
ATTRIBUTE_MODELS = {
|
| 25 |
"English": (
|
| 26 |
-
"unitary/toxic-bert",
|
| 27 |
"distilbert-base-uncased-finetuned-sst-2-english",
|
|
|
|
| 28 |
"cardiffnlp/twitter-roberta-base-sentiment-latest",
|
| 29 |
)
|
| 30 |
}
|
|
|
|
| 23 |
|
| 24 |
ATTRIBUTE_MODELS = {
|
| 25 |
"English": (
|
|
|
|
| 26 |
"distilbert-base-uncased-finetuned-sst-2-english",
|
| 27 |
+
"unitary/toxic-bert",
|
| 28 |
"cardiffnlp/twitter-roberta-base-sentiment-latest",
|
| 29 |
)
|
| 30 |
}
|