Update app.py

app.py

@@ -20,7 +20,7 @@ def classify_audio(filepath):
             label = "normal"
         elif "extrasystole" in label:
             label = "normal"
-        elif "normal" in label:
+        elif "Normal" in label:
             label = "normal"
         outputs[label] = p["score"]
     return outputs
@@ -31,7 +31,7 @@ st.title("Heartbeat Sound Classification")
 # Theme selection
 theme = st.sidebar.selectbox(
     "Select Theme",
-    ["Light Green", "Light Blue", "Dark Green", "Dark Blue"]
+    ["Light Green", "Light Blue"]
 )
 
 # Add custom CSS for styling based on the selected theme
@@ -77,48 +77,7 @@ elif theme == "Light Blue":
         """,
         unsafe_allow_html=True
     )
-elif theme == "Dark Green":
-    st.markdown(
-        """
-        <style>
-        body, .stApp {
-            background-color: #1b5e20; /* Dark green background */
-        }
-        .stApp {
-            color: #a5d6a7; /* Light green text */
-        }
-        .stButton > button, .stFileUpload > div {
-            background-color: #004d40; /* Dark green button and file uploader background */
-            color: white; /* White text */
-        }
-        .stButton > button:hover, .stFileUpload > div:hover {
-            background-color: #00332c; /* Darker green on hover */
-        }
-        </style>
-        """,
-        unsafe_allow_html=True
-    )
-elif theme == "Dark Blue":
-    st.markdown(
-        """
-        <style>
-        body, .stApp {
-            background-color: #0d47a1; /* Dark blue background */
-        }
-        .stApp {
-            color: #bbdefb; /* Light blue text */
-        }
-        .stButton > button, .stFileUpload > div {
-            background-color: #006064; /* Dark blue button and file uploader background */
-            color: white; /* White text */
-        }
-        .stButton > button:hover, .stFileUpload > div:hover {
-            background-color: #004d40; /* Darker blue on hover */
-        }
-        </style>
-        """,
-        unsafe_allow_html=True
-    )
+
 
 # File uploader for audio files
 uploaded_file = st.file_uploader("Upload an audio file", type=["wav", "mp3"])