app.py
CHANGED
|
@@ -47,12 +47,6 @@ TEXT_GENERATION_MODELS = [
|
|
| 47 |
description="Offers precise and detailed text evaluation",
|
| 48 |
type=ModelType.LOCAL,
|
| 49 |
model_path="mistralai/Mistral-7B-v0.1"
|
| 50 |
-
),
|
| 51 |
-
ModelConfig(
|
| 52 |
-
name="Claude-2",
|
| 53 |
-
description="Provides comprehensive toxicity assessment",
|
| 54 |
-
type=ModelType.INFERENCE_API,
|
| 55 |
-
model_id="anthropic/claude-2"
|
| 56 |
)
|
| 57 |
]
|
| 58 |
|
|
@@ -62,24 +56,6 @@ CLASSIFICATION_MODELS = [
|
|
| 62 |
description="Fine-tuned for toxic content detection",
|
| 63 |
type=ModelType.LOCAL,
|
| 64 |
model_path="unitary/toxic-bert"
|
| 65 |
-
),
|
| 66 |
-
ModelConfig(
|
| 67 |
-
name="RoBERTa-Toxic",
|
| 68 |
-
description="Advanced toxic pattern recognition",
|
| 69 |
-
type=ModelType.INFERENCE_API,
|
| 70 |
-
model_id="unitary/multilingual-toxic-xlm-roberta"
|
| 71 |
-
),
|
| 72 |
-
ModelConfig(
|
| 73 |
-
name="DistilBERT-Toxic",
|
| 74 |
-
description="Efficient toxicity classification",
|
| 75 |
-
type=ModelType.LOCAL,
|
| 76 |
-
model_path="unitary/multilingual-toxic-distilbert"
|
| 77 |
-
),
|
| 78 |
-
ModelConfig(
|
| 79 |
-
name="XLM-RoBERTa-Toxic",
|
| 80 |
-
description="Multilingual toxicity detection",
|
| 81 |
-
type=ModelType.INFERENCE_API,
|
| 82 |
-
model_id="unitary/multilingual-toxic-xlm-roberta"
|
| 83 |
)
|
| 84 |
]
|
| 85 |
|
|
|
|
| 47 |
description="Offers precise and detailed text evaluation",
|
| 48 |
type=ModelType.LOCAL,
|
| 49 |
model_path="mistralai/Mistral-7B-v0.1"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 50 |
)
|
| 51 |
]
|
| 52 |
|
|
|
|
| 56 |
description="Fine-tuned for toxic content detection",
|
| 57 |
type=ModelType.LOCAL,
|
| 58 |
model_path="unitary/toxic-bert"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 59 |
)
|
| 60 |
]
|
| 61 |
|