Auto-deploy from GitHub: c09552b1b4b0829d2a857bdf91e7e48ba2bb8eb3

scripts/predict.py

@@ -61,9 +61,16 @@ def predict_pipeline(audio_file, lyrics):
     classifier.model.eval()
 
     # 8.) Run prediction
-    probability, prediction, label = classifier.predict_single(results.flatten())
-
-    return {"probability": probability, "prediction": prediction, "label": label}
+    confidence, prediction, label, probability = classifier.predict_single(
+        results.flatten()
+    )
+
+    return {
+        "confidence": confidence,
+        "prediction": prediction,
+        "label": label,
+        "probability": probability,
+    }
 
 
 if __name__ == "__main__":

src/models/mlp.py

@@ -496,7 +496,7 @@ class MLPClassifier:
 
         confidence = probability * 100 if prediction == 1 else (1 - probability) * 100
 
-        return confidence, prediction, label
+        return confidence, prediction, label, probability
 
     def predict_batch(self, features: np.ndarray, return_details: bool = False) -> Dict:
         """