Update app.py

app.py

@@ -15,11 +15,7 @@ def classify_audio(audio_file_path):
             headers=headers,
             files={"file": audio_file}
         )
-    # Check the response status and return JSON if successful, or an error message
-    if response.status_code == 200:
-        return response.json()
-    else:
-        return {"error": f"Failed to classify audio. Status code: {response.status_code}"}
+    return response.json()
 
 # Streamlit interface
 st.title("Audio Classifier")
@@ -28,12 +24,8 @@ st.title("Audio Classifier")
 audio_folder = "audio_files"
 
 # List the audio files in the folder
-try:
-    audio_files = os.listdir(audio_folder)
-    audio_file_options = [f for f in audio_files if f.endswith(('.mp3', '.wav'))]
-except Exception as e:
-    st.error(f"Error accessing the audio files: {e}")
-    st.stop()
+audio_files = os.listdir(audio_folder)
+audio_file_options = [f for f in audio_files if f.endswith(('.mp3', '.wav'))]
 
 # Dropdown to select an audio file
 selected_file = st.selectbox("Select an audio file:", audio_file_options)
@@ -48,21 +40,6 @@ if st.button("Classify"):
     
     # Get and display the classification results
     results = classify_audio(audio_file_path)
-    
-    # Check for errors
-    if "error" in results:
-        st.error(results["error"])
-    else:
-        st.write("Results:")
-        # Loop through each result in the list of results
-        for result in results:
-            # Check if result is a dictionary with 'label' and 'score' keys
-            if isinstance(result, dict) and 'label' in result and 'score' in result:
-                st.write(f"Label: {result['label']}, Confidence: {result['score']:.2f}")
-            else:
-                st.error(f"Unexpected result format: {result}")
-    # Get and display the classification results
-    results = classify_audio(audio_file_path)
     st.write("Results:")
     for result in results:
         st.write(f"Label: {result['label']}, Confidence: {result['score']:.2f}")