Update predict_utils.py

predict_utils.py

@@ -24,6 +24,7 @@ def predict(text, model, tokenizer):
     with torch.no_grad():
         outputs = model(input_ids, attention_mask)
     probs = outputs.cpu().numpy()[0]
-    result = {label: (f"✅ Ya ({prob:.2f})" if prob >= thr else f"❌ Tidak ({prob:.2f})")
-              for label, prob, thr in zip(LABELS, probs, THRESHOLDS)}
+
+    # Buat dictionary {label: prob} untuk Gradio Label with confidence
+    result = {label: float(prob) for label, prob in zip(LABELS, probs)}
     return result