Use of Tableau color

app.py

@@ -127,14 +127,4 @@ with demo:
         gr.Markdown(article)
         
 
-demo.launch()  
-  
-""" gr.Interface(
-    inference, 
-    gr.inputs.Audio(type="filepath", label="Input"), 
-    [gr.outputs.File(label="Output")],
-    title=title,
-    description=description,
-    article=article,
-    examples=examples,
-    ).launch().queue() """
+demo.launch()

utils.py

@@ -3,6 +3,7 @@ import collections
 
 import pandas as pd
 import matplotlib.pyplot as plt
+import matplotlib.colors as mcolors
 from matplotlib.patches import Rectangle
 from PIL import Image
 
@@ -21,7 +22,7 @@ def sequence_to_pandas_dataframe(sequence):
         pd_dict["end_time"].append(note.end_time)
         pd_dict["duration"].append(note.end_time - note.start_time)
         pd_dict["pitch"].append(note.pitch)
-        pd_dict['instrument'].append(note.instrument)
+        pd_dict["instrument"].append(note.instrument)
 
     return pd.DataFrame(pd_dict)
 
@@ -29,9 +30,8 @@ def dataframe_to_pianoroll_img(df):
     fig = plt.figure(figsize=(8, 5))
     ax = fig.add_subplot(111)
     ax.scatter(df.start_time, df.pitch, c="white")
-    colors = plt.get_cmap("tab20")
-    print(df["instrument"].unique())
-    colordict = {inst: colors[i] for i, inst in enumerate(df["instrument"].unique())}
+    print(mcolors.TABLEAU_COLORS)
+    colordict = {inst: mcolors.TABLEAU_COLORS[i] for i, inst in enumerate(df["instrument"].unique())}
     for _, row in df.iterrows():
         ax.add_patch(Rectangle((row["start_time"],
                                 row["pitch"]-0.4),