Update app.py

app.py

@@ -18,20 +18,26 @@ else:
 # Initialize the audio classification pipeline with the MIT model
 pipe = pipeline("audio-classification", model="MIT/ast-finetuned-audioset-10-10-0.4593")
 
-# Define the function to classify an audio file
+# Define the function to classify an audio file and return the top 3 results
 def classify_audio(audio):
     result = pipe(audio)
-    return {label['label']: label['score'] for label in result}
+    # The pipeline returns a list of dicts sorted by score.
+    # We select the top 3 results here.
+    top_3_results = result[:3]
+    # Convert the list of dicts to a single dictionary for the Label component
+    return {label['label']: label['score'] for label in top_3_results}
 
 # Set up the Gradio interface
+# We removed `num_top_classes=3` from `gr.Label` and instead handle the
+# top-3 logic inside the `classify_audio` function. This avoids the bug.
 app = gr.Interface(
     fn=classify_audio,                  # Function to classify audio
     inputs=gr.Audio(type="filepath"),   # Input for uploading an audio file
-    outputs=gr.Label(num_top_classes=3), # Output with top 3 classification results
+    outputs=gr.Label(),                 # Output Label will display the dictionary from the function
     title="Audio Classification",        # App title
     description="Upload an audio file to classify it using MIT's fine-tuned AudioSet model."
 )
 
 # Launch the app
 if __name__ == "__main__":
-    app.launch()
+    app.launch()