Update app.py

app.py

@@ -3,18 +3,18 @@ from transformers import pipeline
 
 # Define model names
 models = {
-    "ModernBERT Base (gender)": "breadlicker45/ModernBERT-base-gender",
-    "ModernBERT Large (gender)": "breadlicker45/ModernBERT-large-gender"
+    "ModernBERT Slop Classifier v1": "underscore2/modernbert_base_slop_classifier",
+    "ModernBERT Slop Classifier v2": "underscore2/modernbert_base_slop_classifier_v2"
 }
 
 # Define the mapping for user-friendly labels
 # Note: Transformers pipelines often output 'LABEL_0', 'LABEL_1'.
 # We handle potential variations like just '0', '1'.
 label_map = {
-    "LABEL_0": "Male (0)",
-    "0": "Male (0)",
-    "LABEL_1": "Female (1)",
-    "1": "Female (1)"
+    "LABEL_0": "Human (0)",
+    "0": "Human (0)",
+    "LABEL_1": "LLM (1)",
+    "1": "LLM (1)"
 }
 
 # Function to load the selected model and classify text
@@ -48,18 +48,18 @@ interface = gr.Interface(
         gr.Dropdown(
             list(models.keys()),
             label="Select Model",
-            value="ModernBERT Large (gender)" # Default model
+            value="ModernBERT Slop Classifier v2" # Default model
         ),
         gr.Textbox(
             lines=2,
-            placeholder="Enter text to classify for perceived gender...", # Corrected placeholder
+            placeholder="Enter text to classify", # Corrected placeholder
             value="This is an example sentence." # Changed example text
         )
     ],
     # The gr.Label component works well for showing classification scores
     outputs=gr.Label(num_top_classes=2), # Show both classes explicitly
-    title="ModernBERT Gender Classifier",
-    description="Select a model and enter a sentence to see the perceived gender classification (Male=0, Female=1) and confidence scores. Note: Text-based gender classification can be unreliable and reflect societal biases.", # Updated description
+    title="ModernBERT Slop Classifier",
+    description="Enter a sentence to see the slop score and confidence scores", # Updated description
 )
 
 # Launch the app