Update app.py

app.py

@@ -0,0 +1,38 @@
+# Streamlit app code
+import streamlit as st
+from transformers import pipeline
+import os
+
+# Title of the Streamlit app
+st.title('Music Genre Classification')
+
+# File uploader to upload an audio file
+audio_file = st.file_uploader("Upload an audio file", type=["wav", "mp3", "flac"])
+
+if audio_file is not None:
+    # Save the uploaded file to a temporary directory
+    if not os.path.exists("temp_audio"):
+        os.makedirs("temp_audio")
+    
+    with open(os.path.join("temp_audio", audio_file.name), "wb") as f:
+        f.write(audio_file.getbuffer())
+    
+    # Use the uploaded audio file path
+    audio_path = os.path.join("temp_audio", audio_file.name)
+
+    # Display a loading message
+    st.text("Classifying the audio file... Please wait.")
+
+    # Initialize the audio classification pipeline
+    pipe = pipeline("audio-classification", model="sandychoii/distilhubert-finetuned-gtzan-audio-classification")
+    
+    # Perform classification
+    result = pipe(audio_path)
+
+    # Display the results
+    st.subheader("Classification Results:")
+    for label in result:
+        st.write(f"**Genre**: {label['label']} | **Confidence**: {label['score']:.4f}")
+
+    # Option to play the uploaded audio
+    st.audio(audio_file)