Update app.py

app.py

@@ -2,10 +2,10 @@ import gradio as gr
 from transformers import pipeline
 
 # Load models
-text_emotion_model = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta-base", return_all_scores=False)
-audio_emotion_model = pipeline("audio-classification", model="superb/wav2vec2-base-superb-er", return_all_scores=False)
+text_emotion_model = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta-base")
+audio_emotion_model = pipeline("audio-classification", model="superb/wav2vec2-base-superb-er")
 
-# Emoji map
+# Emoji mapping
 emoji_map = {
     "joy": "😊", "happy": "😊", "anger": "😠", "angry": "😠",
     "sadness": "😢", "sad": "😢", "fear": "😨", "calm": "😌",
@@ -13,10 +13,12 @@ emoji_map = {
     "optimism": "🙂"
 }
 
+# Format output
 def format_emotion(label, score):
     emoji = emoji_map.get(label.lower(), "❓")
     return f"{emoji} <b>{label.title()}</b> ({int(score * 100)}%)"
 
+# Main prediction function
 def detect_emotion(mode, text, audio):
     if mode == "Text" and text:
         result = text_emotion_model(text)[0]
@@ -26,30 +28,33 @@ def detect_emotion(mode, text, audio):
         return f"🎧 <b>Audio Emotion:</b><br>{format_emotion(result['label'], result['score'])}"
     return "⚠️ <b>Please provide valid input.</b>"
 
-def clear_inputs():
+# Clear inputs
+def clear_all():
     return "Text", "", None, ""
 
+# Build UI
 with gr.Blocks() as demo:
     gr.Markdown("## 🎭 Emotion Detection App (Text or Audio)")
 
     mode = gr.Radio(["Text", "Audio"], label="Choose Input Mode", value="Text")
 
-    text_input = gr.Textbox(label="💬 Enter your text", visible=True).style(container=True)
-    audio_input = gr.Audio(type="filepath", label="🎤 Record or Upload Audio", visible=False).style(container=True)
+    text_input = gr.Textbox(label="💬 Enter your text", visible=True)
+    audio_input = gr.Audio(type="filepath", label="🎤 Record or Upload Audio", visible=False)
 
-    output = gr.HTML(label="🎯 Output", value="", show_label=True)
+    output = gr.HTML(label="🎯 Output", value="")
 
     detect_btn = gr.Button("🔍 Detect Emotion")
     clear_btn = gr.Button("❌ Clear Inputs")
 
+    # Show/hide inputs based on mode
     def toggle_inputs(mode):
         return (
             gr.update(visible=(mode == "Text")),
             gr.update(visible=(mode == "Audio"))
         )
-
     mode.change(fn=toggle_inputs, inputs=mode, outputs=[text_input, audio_input])
+
     detect_btn.click(fn=detect_emotion, inputs=[mode, text_input, audio_input], outputs=output)
-    clear_btn.click(fn=clear_inputs, inputs=[], outputs=[mode, text_input, audio_input, output])
+    clear_btn.click(fn=clear_all, inputs=[], outputs=[mode, text_input, audio_input, output])
 
 demo.launch()