Update app.py

app.py

@@ -1,8 +1,5 @@
 from transformers import pipeline
 import gradio as gr
-import emoji
-pip install emoji
-
 
 # Load pre-trained emotion classification model
 model_checkpoint = "MuntasirHossain/RoBERTa-base-finetuned-emotion"
@@ -25,20 +22,19 @@ def classify_and_visualize_emotion(text):
     # Get the corresponding emoji for the predicted emotion
     predicted_emoji = emoji_mapping.get(emotion_label, "🤔")
 
-    # Construct the result string with the predicted emotion and emoji
-    result_text = f"Predicted Emotion: {emotion_label.capitalize()} {predicted_emoji}"
-
-    return result_text
+    return f"{emotion_label.capitalize()} {predicted_emoji}"
 
 # Define interface title and description
 title = "Emotion Classifier with Emoji Visualization"
 description = "This AI model classifies text expressions into six emotions: joy, sadness, love, anger, fear, and surprise. The result is visualized with emojis."
 
-# Set up Gradio Interface
-gr.Interface(
+# Set up Gradio Interface with HTML visualization
+iface = gr.Interface(
     fn=classify_and_visualize_emotion,
     inputs="textbox",
-    outputs="html",
+    outputs=gr.Html(),
     title=title,
     description=description,
-).launch()
+)
+
+iface.launch()