Update app.py

app.py

@@ -7,9 +7,13 @@ pipe = pipeline("audio-classification", model=MODEL_ID)
 
 def classify_audio(filepath):
     preds = pipe(filepath)
-    outputs = {}
+    outputs = {"normal": 0.0, "murmur": 0.0, "artifact": 0.0}
     for p in preds:
-        outputs[p["label"].replace('_', ' ')] = p["score"]
+        label = p["label"].replace('_', ' ')
+        if label in outputs:
+            outputs[label] += p["score"]
+        else:
+            outputs["normal"] += p["score"]
     return outputs
 
 # Streamlit app layout