[Fix] default heatmap

app.py

@@ -99,6 +99,9 @@ links_markdown = """
 """
 
 model_init = ["HuggingFaceTB/SmolLM2-1.7B-Instruct", "meta-llama/Llama-3.1-8B-Instruct", "microsoft/phi-4", "google/gemma-2-27b-it", "Qwen/Qwen2.5-32B-Instruct", "meta-llama/Llama-3.3-70B-Instruct"]
+dataset_init = "mmlu_pro"
+metric_init = "CAPA"
+
 
 # Create Gradio interface
 with gr.Blocks(title="LLM Similarity Analyzer") as demo:
@@ -109,7 +112,7 @@ with gr.Blocks(title="LLM Similarity Analyzer") as demo:
         dataset_dropdown = gr.Dropdown(
             choices=get_leaderboard_datasets(model_init),
             label="Select Dataset",
-            value="mmlu_pro",
+            value=dataset_init,
             filterable=True,
             interactive=True,
             allow_custom_value=False,
@@ -118,6 +121,7 @@ with gr.Blocks(title="LLM Similarity Analyzer") as demo:
         metric_dropdown = gr.Dropdown(
             choices=["CAPA", "CAPA (det.)", "Error Consistency"],
             label="Select Metric",
+            value=metric_init,
             info="Select a similarity metric to compute"
         )
 
@@ -138,7 +142,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, dataset_dropdown, metric_dropdown), label="Similarity Heatmap", visible=True)
+    heatmap = gr.Image(value=create_heatmap(model_init, dataset_init, metric_init), label="Similarity Heatmap", visible=True)
 
     generate_btn.click(
         fn=validate_inputs,