Update app.py

app.py

@@ -10,7 +10,17 @@ def classify_audio(filepath):
     preds = pipe(filepath)
     outputs = {}
     for p in preds:
-        outputs[p["label"].replace('_', ' ')] = p["score"]
+        label = p["label"].replace('_', ' ')
+        # Simplify the labels as required
+        if "artifact" in label:
+            label = "artifact"
+        elif "murmur" in label:
+            label = "murmur"
+        elif "extra" in label:
+            label = "normal"
+        elif "normal" in label:
+            label = "normal"
+        outputs[label] = p["score"]
     return outputs
 
 # Streamlit app layout