update

app.py

@@ -120,6 +120,7 @@ def get_folders_matching_format(directory):
 
 
 def get_unique_column_names(all_data):
+
     # column_names = {}
     #
     # for folder_name, files in all_data.items():
@@ -178,10 +179,36 @@ def update_table(period: str,
         if 'Individual Tests' in color_columns:
             target_color_columns.extend([col for col in filtered_data.columns if col not in ['Name', 'Parameters Count (B)', 'Average (lower=better)']])
 
-        styler = filtered_data.style.format(formatter).background_gradient(
+        # styler = filtered_data.style.format(formatter).background_gradient(
+        #     cmap=cmap,
+        #     subset=target_color_columns,
+        #     vmin=min_value,
+        #     vmax=max_value
+        # ).apply(color_column, subset=['Parameters Count (B)'])
+
+        # for better visualization
+        second_largest = filtered_data['Average (lower=better)'].nlargest(2).iloc[-1]
+        min_value = filtered_data['Average (lower=better)'].min().min()
+        if len(models) == 1 and '~7B' in models:
+            max_value = second_largest
+        else:
+            max_value = filtered_data['Average (lower=better)'].max().max()
+
+        specific_column = 'Average (lower=better)'
+
+        styler_specific = filtered_data.style.background_gradient(
             cmap=cmap,
-            subset=target_color_columns
-        ).apply(color_column, subset=['Parameters Count (B)'])
+            subset=[specific_column],
+            vmin=min_value,
+            vmax=max_value
+        )
+
+        styler_other = filtered_data.style.background_gradient(
+            cmap=cmap,
+            subset=[col for col in target_color_columns if col != specific_column]
+        )
+
+        styler = styler_specific.use(styler_other.export()).format(formatter).apply(color_column, subset=['Parameters Count (B)'])
 
         return styler
     else: