[Fix] plot axes and visibility

app.py

@@ -13,7 +13,8 @@ def create_heatmap(selected_models, selected_dataset):
     similarities = (similarities + similarities.T) / 2  # Make symmetric
     
     # Create plot
-    fig = go.Figure(data=go.Heatmap(
+    fig = go.Figure()
+    fig.add_trace(go.Heatmap(
         z=similarities,
         x=selected_models,
         y=selected_models,