[Fix] plot axes and visibility

app.py

@@ -73,7 +73,7 @@ with gr.Blocks(title="LLM Similarity Analyzer") as demo:
     ).then(
         fn=create_heatmap,
         inputs=[model_dropdown, dataset_dropdown],
-        outputs=[heatmap, heatmap.visible]
+        outputs=[heatmap, heatmap]
     )
     
     # Clear button
@@ -83,5 +83,6 @@ with gr.Blocks(title="LLM Similarity Analyzer") as demo:
         outputs=[model_dropdown, dataset_dropdown, heatmap]
     )
 
+
 if __name__ == "__main__":
     demo.launch()