Spaces:

bethgelab
/

lm-similarity

Sleeping

Joschka Strueber commited on Feb 6

Commit

c4145ee

1 Parent(s): bf2618d

[Fix] error in default heatmap

Files changed (1) hide show

app.py CHANGED Viewed

@@ -138,7 +138,7 @@ with gr.Blocks(title="LLM Similarity Analyzer") as demo:
     )
     generate_btn = gr.Button("Generate Heatmap", variant="primary")
-    heatmap = gr.Image(value=create_heatmap(model_init), label="Similarity Heatmap", visible=True)
     generate_btn.click(
         fn=validate_inputs,

     )
     generate_btn = gr.Button("Generate Heatmap", variant="primary")
+    heatmap = gr.Image(value=create_heatmap(model_init, dataset_dropdown, metric_dropdown), label="Similarity Heatmap", visible=True)
     generate_btn.click(
         fn=validate_inputs,