Update app.py

app.py

@@ -223,9 +223,20 @@ class GradioInference:
         formatted_keywords = "\n".join([f"• {keyword}" for keyword in keywords])
 
         progress(0.80, desc="Extracting Sentiment")
+
+        # Define a dictionary to map labels to emojis
+        sentiment_emojis = {
+            "positive": "Positive 👍🏼",
+            "negative": "Negative 👎🏼",
+            "neutral": "Neutral 😶",
+        }
+        
         # Sentiment label    
         label = self.classifier(summary)[0]["label"]
 
+        # Format the label with emojis
+        formatted_sentiment = sentiment_emojis.get(label, label)
+        
         progress(0.90, desc="Generating Wordcloud")
         # WordCloud object
         wordcloud = WordCloud(colormap = "Oranges").generate(
@@ -238,7 +249,7 @@ class GradioInference:
                 results["text"],
                 transcription_summary[0]["summary_text"],
                 formatted_keywords,
-                label,
+                formatted_sentiment,
                 wordcloud_image,
             )
         else:
@@ -246,7 +257,7 @@ class GradioInference:
                 results["text"],
                 summary,
                 formatted_keywords,
-                label,
+                formatted_sentiment,
                 wordcloud_image,
             )