Upload 3 files

app.py

@@ -0,0 +1,77 @@
+import streamlit as st
+from transformers import AutoModelForSequenceClassification, AutoTokenizer
+import torch
+
+# Load pre-trained emotion detection model
+model_name = "j-hartmann/emotion-english-distilroberta-base"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForSequenceClassification.from_pretrained(model_name)
+
+# Get the actual emotion labels from the model
+emotion_labels = model.config.id2label
+
+# Emotion styles (emoji + colors)
+emotion_styles = {
+    "joy": {"emoji": "😃", "color": "#E6E6FA"},        
+    "sadness": {"emoji": "😢", "color": "#3498DB"},    
+    "anger": {"emoji": "😡", "color": "#FFDAB9"},      
+    "fear": {"emoji": "😨", "color": "#FFFACD"},       
+    "surprise": {"emoji": "😲", "color": "#98FB98"},   
+    "disgust": {"emoji": "🤢", "color": "#FFB6C1"},          
+    "neutral": {"emoji": "😐", "color": "#D3D3D3"}    
+}
+
+# Streamlit UI
+st.set_page_config(page_title="Emotion Detection", layout="centered")
+
+# 🎨 Change Background Color
+st.markdown(
+    """
+    <style>
+    body {
+        background-color:#AFEEEE; /* Light Grayish Blue */
+    }
+    </style>
+    """,
+    unsafe_allow_html=True
+)
+
+# 🎯 Add Image at the Top
+st.image("innomatics-footer-logo.webp", use_container_width=True,width=300) 
+
+st.markdown("<h1 style='text-align: center; color: #7D3C98;'>Emotion Detection using ML</h1>", unsafe_allow_html=True)
+st.markdown("<h3 style='text-align: left; color: #3498DB;'>📝 Enter the text:</h3>", unsafe_allow_html=True)
+
+# User Input
+user_text = st.text_input("", placeholder="Type your text here...")
+
+if st.button("Submit"):
+    if user_text:
+        # Tokenize input text
+        inputs = tokenizer(user_text, return_tensors="pt")
+
+        # Get model predictions
+        with torch.no_grad():
+            outputs = model(**inputs)
+
+        # Get the predicted emotion
+        scores = outputs.logits[0]
+        predicted_label_id = torch.argmax(scores).item()
+        predicted_emotion = emotion_labels[predicted_label_id].strip().lower()
+
+        # Get emoji & color
+        emotion_data = emotion_styles.get(predicted_emotion, {"emoji": "😐", "color": "#D3D3D3"})
+        emoji_display = emotion_data["emoji"]
+        text_color = emotion_data["color"]
+
+        # Display Results with Color
+        st.markdown(
+            f"""
+            <div style="text-align: center; padding: 10px; border-radius: 10px; background-color: {text_color}; color: black; font-size: 24px;">
+                <b>Detected Emotion:</b> {predicted_emotion.capitalize()} {emoji_display}
+            </div>
+            """,
+            unsafe_allow_html=True
+        )
+    else:
+        st.warning("Please enter some text!")

requirements.txt

@@ -0,0 +1,3 @@
+streamlit 
+transformer
+torch