Update app.py

app.py

@@ -43,7 +43,9 @@ def predict_label(text_input):
 
     X = vectorizer.transform([text_input])
     prediction = model.predict(X)[0]
-    return f"🔮 Prediction: {prediction}"
+    proba = model.predict_proba(X).max()  # highest confidence for predicted class
+
+    return f"🔮 Prediction: {prediction} (confidence: {proba:.2%})"
 
 with gr.Blocks() as demo:
     gr.Markdown("# 🧠 Text Classification App")