Joschka Strueber commited on
Commit
45b2347
·
1 Parent(s): 9e1c5ed

[Add] default heatmap

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -138,8 +138,8 @@ with gr.Blocks(title="LLM Similarity Analyzer") as demo:
138
  )
139
 
140
  generate_btn = gr.Button("Generate Heatmap", variant="primary")
141
- heatmap = gr.Image(label="Similarity Heatmap", visible=True)
142
-
143
  generate_btn.click(
144
  fn=validate_inputs,
145
  inputs=[model_dropdown, dataset_dropdown],
 
138
  )
139
 
140
  generate_btn = gr.Button("Generate Heatmap", variant="primary")
141
+ heatmap = gr.Image(value=create_heatmap(model_init), label="Similarity Heatmap", visible=True)
142
+
143
  generate_btn.click(
144
  fn=validate_inputs,
145
  inputs=[model_dropdown, dataset_dropdown],