Update app.py

app.py

@@ -10,7 +10,7 @@ def classify_audio(filepath):
     preds = pipe(filepath)
     outputs = {}
     for p in preds:
-        outputs[p["label"]] = p["score"]
+        outputs[p["label"].replace('_', ' ')] = p["score"]
     return outputs
 
 # Streamlit app layout
@@ -43,10 +43,11 @@ if uploaded_file is not None:
 st.write("Examples:")
 examples = ['normal.wav', 'murmur.wav', 'extra_systole.wav', 'extra_hystole.wav', 'artifact.wav']
 for example in examples:
-    st.write(example)
-    audio_bytes = open(example, 'rb').read()
-    st.audio(audio_bytes, format='audio/wav')
-    results = classify_audio(example)
-    st.write("Results:")
-    results_str = "\n".join([f"{label}: {score:.2f}" for label, score in results.items()])
-    st.text(results_str)
+    if st.button(example):
+        st.subheader(f"Example: {example}")
+        audio_bytes = open(example, 'rb').read()
+        st.audio(audio_bytes, format='audio/wav')
+        results = classify_audio(example)
+        st.write("Results:")
+        results_str = "\n".join([f"{label}: {score:.4f}" for label, score in results.items()])
+        st.text(results_str)