Update app.py

app.py

@@ -8,10 +8,16 @@ pipe = pipeline("audio-classification", model=MODEL_ID)
 
 def classify_audio(filepath):
     preds = pipe(filepath)
-    outputs = {}
+    outputs = {"normal": 0.0, "artifact": 0.0, "murmur": 0.0}
     for p in preds:
-        label = p["label"].replace('_', ' ')
-        outputs[label] = p["score"]
+        label = p["label"]
+        # Simplify the labels and accumulate the scores
+        if "artifact" in label:
+            outputs["artifact"] += p["score"]
+        elif "murmur" in label:
+            outputs["murmur"] += p["score"]
+        elif "extra" in label or "normal" in label:
+            outputs["normal"] += p["score"]
     return outputs
 
 # Streamlit app layout