Heatmap: Second heatmap added

server.py

@@ -840,13 +840,39 @@ class LeaderboardServer:
         sorted_indices = sizes_series.sort_values(ascending=False).index
         original_scores = original_scores.loc[sorted_indices]  # Sort rows by model size
         
-        fig = create_heatmap(
-            original_scores,
-            original_scores * 100,
+        # Smaller models
+        original_scores_sub = original_scores[sizes_series < 16]
+
+        # Apply quantile transformation independently for each row
+        normalized_scores_sub = original_scores_sub.apply(lambda x: (x - x.min()) / (x.max() - x.min()), axis=0)
+
+        # Call the heatmap function with the normalized data
+        p1 = create_heatmap(
+            normalized_scores_sub,
+            original_scores_sub * 100,
+            x_axis_label="Model < 16B",
             y_axis_label=fig_y_axis_label,
         )
         
-        return fig
+        # Bigger models
+        original_scores_sub = original_scores[sizes_series >= 16]
+
+        # Apply quantile transformation independently for each row
+        normalized_scores_sub = original_scores_sub.apply(lambda x: (x - x.min()) / (x.max() - x.min()), axis=0)
+
+        # Call the heatmap function with the normalized data
+        p2 = create_heatmap(
+            normalized_scores_sub,
+            original_scores_sub * 100,
+            x_axis_label="Model >= 16B",
+            y_axis_label=fig_y_axis_label,
+            y_axis_visible=False,
+        )
+        
+        from bokeh.layouts import row
+        layout = row(p1, p2)
+        
+        return layout
 
     def get_leaderboard_csv(self, pre_submit=None, category=None, kind_of_p_value=None):
         if pre_submit == None: