[Fix] heatmap not generated and deselection causes error

app.py

@@ -30,7 +30,7 @@ def create_heatmap(selected_models, selected_dataset):
     )
     
     print("Heatmap generated successfully")  # Debugging
-    return gr.Plot.update(value=fig, visible=True)
+    return fig
 
 def validate_inputs(selected_models, selected_dataset):
     if not selected_models:
@@ -76,7 +76,7 @@ with gr.Blocks(title="LLM Similarity Analyzer") as demo:
     # Clear button
     clear_btn = gr.Button("Clear Selection")
     clear_btn.click(
-        lambda: [None, None, gr.Plot.update(visible=False)],
+        lambda: [None, None, None],
         outputs=[model_dropdown, dataset_dropdown, heatmap]
     )