custom colormap because the previous one is annoying

app.py

@@ -8,6 +8,7 @@ from transformers import AutoModelForSequenceClassification
 from transformers import AutoTokenizer
 import gradio as gr
 import matplotlib.pyplot as plt
+from matplotlib.colors import LinearSegmentedColormap
 import plotly.express as px
 import seaborn as sns
 
@@ -90,11 +91,13 @@ def prepare_heatmap_data(data):
     return heatmap_data
 
 
-def plot_emotion_heatmap(heatmap_data, cmap="Greys_r"):
+def plot_emotion_heatmap(heatmap_data):
     heatmap_data = heatmap_data.T
 
     fig = plt.figure(figsize=(len(heatmap_data.columns) * 0.5 + 4, len(heatmap_data.index) * 0.5 + 2))
 
+    cmap = LinearSegmentedColormap.from_list("white_to_grey", ["#ffffff", "#aaaaaa"])
+
     sns.heatmap(
         heatmap_data,
         annot=False,