Update app.py

app.py

@@ -55,7 +55,11 @@ def custom_sentiment(text):
         outputs = sentiment_model(**inputs)
     probs = torch.nn.functional.softmax(outputs.logits, dim=1)
     label_idx = torch.argmax(probs).item()
-    label = sentiment_model.config.id2label[label_idx]
+    
+    # Map index to custom label
+    label_map = {0: "supportive", 1: "undermining"}
+    label = label_map[label_idx]
+    
     score = probs[0][label_idx].item()
     return {"label": label, "score": score}