Fix unmatched labels number.

app.py

@@ -16,8 +16,12 @@ def plot_bar(value_counts):
 def plot_confusion_matrix(y_pred, y_true):
     cm = confusion_matrix(y_true, y_pred, normalize='true')
     fig, ax = plt.subplots(figsize=(6, 6))
+    labels = []
+    for label in SENTI_MAPPING.keys():
+        if (label in y_pred.values) or (label in y_true.values):
+            labels.append(label)
     disp = ConfusionMatrixDisplay(confusion_matrix=cm,
-                                  display_labels=['negative', 'neutral', 'positive'])
+                                  display_labels=labels)
     disp.plot(cmap="Blues", values_format=".2f", ax=ax, colorbar=False)
     plt.title("Normalized Confusion Matrix")
     return fig