breadlicker45 commited on
Commit
0f21896
·
verified ·
1 Parent(s): c4b34d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,8 +3,8 @@ from transformers import pipeline
3
 
4
  # Define model names
5
  models = {
6
- "ModernBERT Base (Go-Emotions)": "cirimus/modernbert-base-go-emotions",
7
- "ModernBERT Large (Go-Emotions)": "cirimus/modernbert-large-go-emotions"
8
  }
9
 
10
  # Function to load the selected model and classify text
 
3
 
4
  # Define model names
5
  models = {
6
+ "ModernBERT Base (gender)": "breadlicker45/ModernBERT-base-gender",
7
+ "ModernBERT Large (gender)": "breadlicker45/ModernBERT-large-gender"
8
  }
9
 
10
  # Function to load the selected model and classify text