Update app.py

app.py

@@ -30,15 +30,21 @@ def classify_confidence(text):
         label = id2label[predicted_class]
         prediction = label_map[label]
         score = round(probs[0][predicted_class].item() * 100, 2)
+        color = "green" if label == "LABEL_0" else "red"
 
-    return f"{prediction} ({score}%)"
+  return f"<b>Prediction:</b> <span style='color: {color}; font-weight: bold'>{label_text}</span> ({score}%)"
 
 iface = gr.Interface(
     fn=classify_confidence,
     inputs=gr.Textbox(lines=3, placeholder="Enter a statement..."),
     outputs="text",
-    title="Confidence Classifier",
-    description="Enter a statement and the model will classify it as 'confident' or 'not confident'.",
+    title="🌟 Confidence Classifier",
+    description="""
+Enter a statement, and this model will predict whether it shows confidence or not.
+
+🔍 <i>Built using a fine-tuned DistilBERT model</i>  
+💡 <i>Great for journaling, speech prep, or daily reflections</i>
+""",
 )
 
 iface.launch()