[Fix] plot axes and visibility

app.py

@@ -5,7 +5,7 @@ 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 gr.Plot.update(visible=False)  # Hide plot
+        return None, gr.update(visible=False)  # Hide plot
     
     # Generate random similarity matrix
     size = len(selected_models)
@@ -33,7 +33,7 @@ def create_heatmap(selected_models, selected_dataset):
         margin=dict(b=100, l=100)  # Add bottom/left margin for labels
     )
     
-    return gr.Plot.update(value=fig, visible=True)
+    return fig, gr.update(visible=True)
 
 
 def validate_inputs(selected_models, selected_dataset):
@@ -74,7 +74,7 @@ with gr.Blocks(title="LLM Similarity Analyzer") as demo:
     ).then(
         fn=create_heatmap,
         inputs=[model_dropdown, dataset_dropdown],
-        outputs=heatmap
+        outputs=[heatmap, heatmap.visible]
     )
     
     # Clear button