Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,7 +4,8 @@ from transformers import pipeline
|
|
| 4 |
# Define model names
|
| 5 |
models = {
|
| 6 |
"ModernBERT Slop Classifier v1": "underscore2/modernbert_base_slop_classifier",
|
| 7 |
-
"ModernBERT Slop Classifier v2": "underscore2/modernbert_base_slop_classifier_v2"
|
|
|
|
| 8 |
}
|
| 9 |
|
| 10 |
# Define the mapping for user-friendly labels
|
|
@@ -48,7 +49,7 @@ interface = gr.Interface(
|
|
| 48 |
gr.Dropdown(
|
| 49 |
list(models.keys()),
|
| 50 |
label="Select Model",
|
| 51 |
-
value=
|
| 52 |
),
|
| 53 |
gr.Textbox(
|
| 54 |
lines=2,
|
|
|
|
| 4 |
# Define model names
|
| 5 |
models = {
|
| 6 |
"ModernBERT Slop Classifier v1": "underscore2/modernbert_base_slop_classifier",
|
| 7 |
+
"ModernBERT Slop Classifier v2": "underscore2/modernbert_base_slop_classifier_v2",
|
| 8 |
+
"ModernBert Large Slop Classifier v3 (Best accuracy)": "underscore2/modernbert_large_slop_classifier_v3"
|
| 9 |
}
|
| 10 |
|
| 11 |
# Define the mapping for user-friendly labels
|
|
|
|
| 49 |
gr.Dropdown(
|
| 50 |
list(models.keys()),
|
| 51 |
label="Select Model",
|
| 52 |
+
value=models[2] # Default model
|
| 53 |
),
|
| 54 |
gr.Textbox(
|
| 55 |
lines=2,
|