Improve confidence score formatting

app.py

@@ -8,16 +8,24 @@ MODEL_PATH = "models/best_model.pkl"
 model = joblib.load(MODEL_PATH)
 
 
+def softmax(scores):
+    scores = np.array(scores)
+    exp_scores = np.exp(scores - np.max(scores))
+    return exp_scores / np.sum(exp_scores)
+
+
 def classify_document(text):
     if not text or len(text.strip()) < 5:
         return "Please enter at least 5 characters.", 0.0
 
     prediction = model.predict([text])[0]
 
-    decision_scores = model.decision_function([text])
-    confidence_score = float(np.max(decision_scores))
+    decision_scores = model.decision_function([text])[0]
+    probabilities = softmax(decision_scores)
+
+    confidence_score = float(np.max(probabilities)) * 100
 
-    return prediction, round(confidence_score, 4)
+    return prediction, round(confidence_score, 2)
 
 
 demo = gr.Interface(
@@ -29,7 +37,7 @@ demo = gr.Interface(
     ),
     outputs=[
         gr.Textbox(label="Predicted Category"),
-        gr.Number(label="Confidence Score")
+        gr.Number(label="Confidence Score (%)")
     ],
     title="BBC News Document Classifier",
     description=(
@@ -39,7 +47,9 @@ demo = gr.Interface(
     examples=[
         ["The football team won the final match after scoring two goals."],
         ["The company reported strong profits and growth in global markets."],
-        ["New software updates improve artificial intelligence performance."]
+        ["New software updates improve artificial intelligence performance."],
+        ["The government introduced a new policy during the parliamentary session."],
+        ["The actor received praise for her performance in the award-winning film."]
     ]
 )