underscore2 commited on
Commit
63a8066
·
verified ·
1 Parent(s): 5bbff19

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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="ModernBERT Slop Classifier v2" # Default model
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,