Spaces:
Running
Running
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,
|