update plot

app.py

@@ -267,7 +267,7 @@ def optimize_model(model, sparsity, context, criteria):
 def create_size_comparison_plot(before_results, after_results, metrics):
     sns.set_style("darkgrid")
     # Increase figure size height to accommodate labels better
-    fig = plt.figure(figsize=(10, 6), dpi=150)
+    fig = plt.figure(figsize=(12, 7), dpi=150)
     fig.patch.set_alpha(0.0)
     ax = plt.gca()
     ax.patch.set_alpha(0.0)
@@ -307,17 +307,17 @@ def create_size_comparison_plot(before_results, after_results, metrics):
         plt.text(bar.get_x() + bar.get_width()/2., height + offset,
                  format_value(height, metrics),
                  ha='center', va='bottom',
-                 fontsize=11,
+                 fontsize=15,
                  fontweight='bold',
                  color='white')
     compression_ratio = ((before_results - after_results) / before_results) * 100 if before_results else 0
     plt.title(f'Model Compression: {compression_ratio:.1f}%', 
-              fontsize=14, 
+              fontsize=18, 
               fontweight='bold', 
               pad=20,
               color='white')
-    plt.xlabel('Model Version', fontsize=12, fontweight='bold', labelpad=10, color='white')
-    plt.ylabel(unit_label_map.get(metrics, metrics), fontsize=12, fontweight='bold', labelpad=10, color='white')
+    plt.xlabel('Model Version', fontsize=15, fontweight='bold', labelpad=10, color='white')
+    plt.ylabel(unit_label_map.get(metrics, metrics), fontsize=15, fontweight='bold', labelpad=10, color='white')
     ax.grid(alpha=0.2, color='gray')
     sns.despine()
     # Use scientific notation for small Energy/Emissions values
@@ -330,13 +330,18 @@ def create_size_comparison_plot(before_results, after_results, metrics):
     plt.ylim(0, max_value * 1.3) # Increased upper limit
     plt.yticks(np.linspace(0, max_value * 1.3, 10))
     ax.tick_params(colors='white')
+    ax.tick_params(axis='x', colors='white', labelsize=16)
+    ax.tick_params(axis='y', colors='white', labelsize=15)
+    for tick_label in ax.get_xticklabels():
+        tick_label.set_fontweight('bold')
     for spine in ax.spines.values():
         spine.set_color('white')
     ax.xaxis.label.set_color('white')
     ax.yaxis.label.set_color('white')
     ax.tick_params(axis='x', colors='white')
     ax.tick_params(axis='y', colors='white')
-    ax.yaxis.set_major_formatter(plt.FuncFormatter(lambda x, p: f'{x:.1f}'))
+    if metrics not in ('Energy', 'Emissions'):
+        ax.yaxis.set_major_formatter(plt.FuncFormatter(lambda x, p: f'{x:.1f}'))
     plt.tight_layout(pad=3.5) # Increased padding from 2.5 to 3.5
     return fig