Update app.py

app.py

@@ -34,18 +34,18 @@ def classify_audio(filepath):
 
 title = "🎵 Music Genre Classifier"
 description = """
-demo to showcase the music
-classification model that we just trained on the [GTZAN](https://huggingface.co/datasets/marsyas/gtzan)
+Music Genre Classifier model (Fine-tuned "ntu-spml/distilhubert") Dataset: [GTZAN](https://huggingface.co/datasets/marsyas/gtzan)
 """
 
-filenames = ['blues.00098.wav', "disco.00020.wav", "metal.00014.wav", "reggae.00021.wav", "rock.00058.wav"]
+filenames = ['rock-it-21275.mp3']
 filenames = [[f"./{f}"] for f in filenames]
 demo = gr.Interface(
     fn=classify_audio,
     inputs=gr.Audio(type="filepath"),
-    outputs=gr.outputs.Label(),
+    outputs=[gr.outputs.Label(), gr.Number(label="Prediction time (s)")],
     title=title,
     description=description,
     examples=filenames,
 )
+
 demo.launch()