[Fix] plot axes and visibility

app.py

@@ -74,14 +74,14 @@ with gr.Blocks(title="LLM Similarity Analyzer") as demo:
     ).then(
         fn=create_heatmap,
         inputs=[model_dropdown, dataset_dropdown],
-        outputs=[heatmap, gr.update(visible=True)]
+        outputs=[heatmap, heatmap.visible])]
     )
     
     # Clear button
     clear_btn = gr.Button("Clear Selection")
     clear_btn.click(
         lambda: [None, None, None],
-        outputs=[model_dropdown, dataset_dropdown, heatmap, gr.update(visible=False)]
+        outputs=[model_dropdown, dataset_dropdown, heatmap]
     )
 
 if __name__ == "__main__":