Update app.py

app.py

@@ -17,13 +17,6 @@ pipe = pipeline("audio-classification", model=model_id, device=device)
 
 
 def classify_audio(filepath):
-    """
-    Goes from
-    [{'score': 0.8339303731918335, 'label': 'country'},
-    {'score': 0.11914275586605072, 'label': 'rock'},]
-    to
-    {"country":  0.8339303731918335, "rock":0.11914275586605072}
-    """
     import time
     start_time = time.time()
     
@@ -39,6 +32,7 @@ def classify_audio(filepath):
     
     return outputs, prediction_time
 
+
 title = "🎵 Music Genre Classifier"
 description = """
 Music Genre Classifier model (Fine-tuned "ntu-spml/distilhubert") Dataset: [GTZAN](https://huggingface.co/datasets/marsyas/gtzan)
@@ -46,10 +40,11 @@ Music Genre Classifier model (Fine-tuned "ntu-spml/distilhubert") Dataset: [GTZA
 
 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(), gr.Number(label="Prediction time (s)")],
+    outputs=[gr.Label(), gr.Number(label="Prediction time (s)")],  # Using updated component names
     title=title,
     description=description,
     examples=[(f,) for f in filenames],