[Fix] make plot visible

app.py

@@ -8,7 +8,7 @@ def create_heatmap(selected_models, selected_dataset):
     print(f"Selected Dataset: {selected_dataset}")  # Debugging
     if not selected_models or not selected_dataset:
         print("No models or dataset selected")  # Debugging
-        return None, False # Hide the plot if no models or dataset selected
+        return None, gr.update(visible=False)   # Hide the plot if no models or dataset selected
     
     # Generate random similarity matrix
     size = len(selected_models)
@@ -30,7 +30,7 @@ def create_heatmap(selected_models, selected_dataset):
     )
     
     print("Heatmap generated successfully")  # Debugging
-    return fig, True
+    return fig, gr.update(visible=True)  # Show the plot
 
 def validate_inputs(selected_models, selected_dataset):
     if not selected_models:
@@ -70,7 +70,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