Update app.py

app.py

@@ -15,9 +15,14 @@ def predict(text):
     for result in all_scores:
         label = result['label']
         score = result['score']
-        if label in ["Politeness", "Sentiment", "Formality", "Toxicity"]:
-            # Highlight only the label
+        if label in ["Politeness"]:
             response += f"<span style='background-color:green; color:black;'>{label}</span>: {score:.3f}<br>"
+        if label in ["Sentiment"]:
+            response += f"<span style='background-color:yellow; color:black;'>{label}</span>: {score:.3f}<br>"
+        if label in ["Formality"]:
+            response += f"<span style='background-color:blue; color:black;'>{label}</span>: {score:.3f}<br>"
+        if label in ["Toxicity"]:
+            response += f"<span style='background-color:red; color:black;'>{label}</span>: {score:.3f}<br>"
         else:
             response += f"{label}: {score:.3f}<br>"