Update app.py

app.py

@@ -7,8 +7,8 @@ models = {
     "ModernBERT Large (Go-Emotions)": "cirimus/modernbert-large-go-emotions"
 }
 
-# Function to load the selected model
-def classify_text(text, model_name):
+# Function to load the selected model and classify text
+def classify_text(model_name, text):
     classifier = pipeline("text-classification", model=models[model_name], top_k=None)
     predictions = classifier(text)
     return {pred["label"]: pred["score"] for pred in predictions[0]}
@@ -29,7 +29,7 @@ interface = gr.Interface(
         )
     ],
     outputs=gr.Label(num_top_classes=5),
-    title="ModernBERT Emotion Classifier",
+    title="Emotion Classifier",
     description="Select a model and enter a sentence to see its associated emotions and confidence scores.",
 )