Update app.py

app.py

@@ -4,12 +4,13 @@ from transformers import pipeline
 classifier = pipeline("audio-classification", model="Wiam/baby-cry-classification-finetuned-babycry-v4")
 
 def classify_baby_cry(audio):
-    return classifier(audio)
+    results = classifier(audio)
+    return {res["label"]: res["score"] for res in results}
 
 gr.Interface(
     fn=classify_baby_cry,
     inputs=gr.Audio(type="filepath"),
-    outputs="label",
+    outputs=gr.Label(num_top_classes=3),
     title="זיהוי בכי תינוקות",
     description="העלה קובץ קול של בכי תינוקות ונדע לזהות מה הוא רוצה"
 ).launch()