Update app.py

app.py

@@ -20,37 +20,26 @@ pipe = pipeline(
 
 
 # --- Core Logic Function ---
-def classify_audio(audio_filepath):
+def classify_audio(audio):
     """
     Classifies the audio, takes the top 3 predictions,
     and formats them into a single, human-readable string.
     """
-    if audio_filepath is None:
+    if audio is None:
         return "Please upload an audio file first."
 
-    preds = pipe(audio_filepath)
-
-    # Format the output as a string instead of a dictionary
-    # This is the key change to fix the TypeError
-    output_str = ""
-    for i, pred in enumerate(preds[:3]):
-        label = pred["label"]
-        score = pred["score"]
-        output_str += f"{i+1}. {label}: {score:.4f}\n"
-
-    return output_str.strip()
-
+    result = pipe(audio)
+    return {label['label']: label['score'] for label in result}
 
 # --- Gradio Interface ---
 # Create the Gradio app interface
 app = gr.Interface(
     fn=classify_audio,
     inputs=gr.Audio(type="filepath", label="Upload Audio File"),
-    outputs=gr.Label(label="Top 3 Predictions"), # This will now receive a simple string
+    outputs=gr.Label(num_top_classes=3), # This will now receive a simple string
     title="Audio Classification with MIT/AST",
     description=(
         "Upload an audio file to classify it. The model will identify the top 3 most likely sound categories. "
-        "This version is corrected to avoid common Gradio backend errors."
     ),
     cache_examples=False,
 )