Update app.py

app.py

@@ -11,8 +11,12 @@ description = "Gradio demo for finetuned Wav2Vec2 model on a custom dataset to p
 
 
 def classify_audio(audio):
-    predictions = pipe(audio)
-    return predictions
+    preds = pipe(audio)
+    outputs = {}
+    for p in preds:
+        outputs[p["label"]] = p["score"]
+    return outputs
+
 
     
 iface = gr.Interface(