[Fix] plot axes and visibility

app.py

@@ -5,20 +5,19 @@ from src.dataloading import get_leaderboard_models_cached, get_leaderboard_datas
 
 def create_heatmap(selected_models, selected_dataset):
     if not selected_models or not selected_dataset:
-        return None, False  # Hide plot
+        return None, gr.update(visible=False)   # Hide the plot if no models or dataset selected
     
     # Generate random similarity matrix
     size = len(selected_models)
     similarities = np.random.rand(size, size)
     similarities = (similarities + similarities.T) / 2  # Make symmetric
     
-    # Create plot with proper axis labels
+    # Create plot
     fig = go.Figure(data=go.Heatmap(
         z=similarities,
         x=selected_models,
         y=selected_models,
-        colorscale='Viridis',
-        hoverongaps=False
+        colorscale='Viridis'
     ))
     
     # Improve axis readability
@@ -32,8 +31,8 @@ def create_heatmap(selected_models, selected_dataset):
         height=800 + 20*len(selected_models),
         margin=dict(b=100, l=100)  # Add bottom/left margin for labels
     )
-    
-    return fig, True
+
+    return fig, gr.update(visible=True)  # Show the plot
 
 
 def validate_inputs(selected_models, selected_dataset):