Joschka Strueber commited on
Commit
f8fdd68
·
1 Parent(s): 8ad245f

[Fix] populate plot

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -5,7 +5,7 @@ from src.dataloading import get_leaderboard_models_cached, get_leaderboard_datas
5
 
6
  def create_heatmap(selected_models, selected_dataset):
7
  if not selected_models or not selected_dataset:
8
- return None, gr.update(visible=False) # Hide if no selection
9
 
10
  # Generate random similarity matrix
11
  size = len(selected_models)
@@ -29,7 +29,7 @@ def create_heatmap(selected_models, selected_dataset):
29
  yaxis_title="Models"
30
  )
31
 
32
- return fig, gr.update(visible=True) # ✅ Ensure visibility
33
 
34
  def validate_inputs(selected_models, selected_dataset):
35
  if not selected_models:
@@ -69,7 +69,7 @@ with gr.Blocks(title="LLM Similarity Analyzer") as demo:
69
  ).then(
70
  fn=create_heatmap,
71
  inputs=[model_dropdown, dataset_dropdown],
72
- outputs=[heatmap, heatmap.visible]
73
  )
74
 
75
  # Clear button
 
5
 
6
  def create_heatmap(selected_models, selected_dataset):
7
  if not selected_models or not selected_dataset:
8
+ return None
9
 
10
  # Generate random similarity matrix
11
  size = len(selected_models)
 
29
  yaxis_title="Models"
30
  )
31
 
32
+ return fig
33
 
34
  def validate_inputs(selected_models, selected_dataset):
35
  if not selected_models:
 
69
  ).then(
70
  fn=create_heatmap,
71
  inputs=[model_dropdown, dataset_dropdown],
72
+ outputs=[heatmap]
73
  )
74
 
75
  # Clear button