Update app.py

app.py

@@ -86,7 +86,7 @@ def yolov9_inference(model_id, img_path=None, vid_path=None, tracking_algorithm
         fig, ax = plt.subplots(figsize=(10, 6))
         #for label in labels:
             #df_label = frame_counts_df[frame_counts_df['label'] == label]
-        sns.barplot(ax = ax, data = df, x = 'label', y = 'count', palette='viridis', hue = 'label', legend = False)
+        sns.barplot(ax = ax, data = df, x = 'label', y = 'count', palette=palette, hue = 'label', legend = False)
 
         # Customizations
         ax.set_title('Count of Labels', fontsize=20)
@@ -104,7 +104,7 @@ def yolov9_inference(model_id, img_path=None, vid_path=None, tracking_algorithm
         fig, ax = plt.subplots(figsize=(10, 6))
         #for label in labels:
             #df_label = frame_counts_df[frame_counts_df['label'] == label]
-        sns.lineplot(ax = ax, data = frame_counts_df,  x = 'frame', y = 'count', hue = 'label')
+        sns.lineplot(ax = ax, data = frame_counts_df,  x = 'frame', y = 'count', hue = 'label', palette = palette)
 
         ax.set_xlabel('Frame')
         ax.set_ylabel('Count')