Update app.py

app.py

@@ -13,32 +13,32 @@ classifier = pipeline(
     model="MennatullahHany/Abert"
 )
 
+label_map = {
+    "LABEL_0": "Safe",
+    "LABEL_1": "Danger"
+}
+
 def predict(audio):
-    try:
-        # تحويل الصوت إلى نص
-        transcription = whisper(audio)["text"]
 
-        # تشغيل موديلك
-        result = classifier(transcription)
+    # تحويل الصوت إلى نص
+    transcription = whisper(audio)["text"]
 
-        # يظهر في Logs
-        print(result)
+    # تشغيل الموديل
+    result = classifier(transcription)[0]
 
-        return {
-            "transcription": transcription,
-            "model_output": result
-        }
+    label = label_map[result["label"]]
 
-    except Exception as e:
-        return {
-            "error": str(e)
-        }
+    return {
+        "emotion": label,
+        "confidence": round(float(result["score"]), 4)
+    }
 
 demo = gr.Interface(
     fn=predict,
     inputs=gr.Audio(type="filepath", label="Upload Audio"),
     outputs=gr.JSON(label="Prediction"),
-    title="Voice Emotion Detection"
+    title="Voice Emotion Detection",
+    description="Upload an audio file to detect whether it is Safe or Danger."
 )
 
 demo.launch()