[Fix] rendering issues

app.py

@@ -3,7 +3,7 @@ import plotly.graph_objects as go
 import numpy as np
 from src.dataloading import get_leaderboard_models_cached, get_leaderboard_datasets
 
-# Force the Plotly renderer to use an iframe-based output
+# Force Plotly to use the iframe renderer
 import plotly.io as pio
 pio.renderers.default = "iframe"
 
@@ -35,7 +35,6 @@ def create_heatmap(selected_models, selected_dataset):
         margin=dict(l=100, r=100, t=100, b=100)
     )
     
-    # (Optional) Force categorical ordering explicitly
     fig.update_xaxes(
         type="category",
         categoryorder="array",
@@ -50,8 +49,8 @@ def create_heatmap(selected_models, selected_dataset):
         automargin=True
     )
     
-    # Return a fully serializable dictionary
-    return fig.to_dict()
+    # Return the Plotly figure object directly
+    return fig
 
 def validate_inputs(selected_models, selected_dataset):
     if not selected_models:
@@ -98,4 +97,5 @@ with gr.Blocks(title="LLM Similarity Analyzer") as demo:
     )
 
 if __name__ == "__main__":
+    # Disable SSR to force client-side rendering on Spaces
     demo.launch(ssr_mode=False)