Update app.py

app.py

@@ -89,9 +89,9 @@ def yolov9_inference(model_id, img_path=None, vid_path=None, tracking_algorithm
         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)
-        ax.set_xlabel('Label', fontsize=15)
-        ax.set_ylabel('Count', fontsize=15)
+        ax.set_title('Number of Objects', fontsize=20)
+        ax.set_xlabel('Object Class', fontsize=15)
+        ax.set_ylabel('Object Count', fontsize=15)
         ax.tick_params(axis='x', rotation=45)  # Rotate x-axis labels for better readability
         sns.despine()  # Remove the top and right spines from plot
         #ax.legend()
@@ -104,11 +104,11 @@ 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', palette = palette)
+        sns.lineplot(ax = ax, data = frame_counts_df[::30],  x = 'frame', y = 'count', hue = 'label', palette = palette)
 
-        ax.set_xlabel('Frame')
-        ax.set_ylabel('Count')
-        ax.set_title('Count of Labels over Frames')
+        ax.set_xlabel('Second')
+        ax.set_ylabel('Object Count')
+        ax.set_title('Number of Objects over Seconds')
         ax.legend()
         ax.grid(True)
         ax.set_facecolor('#D3D3D3')