[Add, Fix] fix clearing model list, improve axes labels

app.py

@@ -13,6 +13,10 @@ def create_heatmap(selected_models, selected_dataset):
     if not selected_models or not selected_dataset:
         return None
     
+    selected_models = selected_models.sort()
+    selected_models_short = [model.split("/")[-1] for model in selected_models]
+    
+    # Generate random similarity matrix
     size = len(selected_models)
     similarities = np.random.rand(size, size)
     similarities = (similarities + similarities.T) / 2
@@ -27,14 +31,14 @@ def create_heatmap(selected_models, selected_dataset):
         cmap="viridis",
         vmin=0,
         vmax=1,
-        xticklabels=selected_models,
-        yticklabels=selected_models
+        xticklabels=selected_models_short,
+        yticklabels=selected_models_short
     )
     
     # Customize plot
-    plt.title(f"Similarity Matrix for {selected_dataset}", fontsize=14)
-    plt.xlabel("Models")
-    plt.ylabel("Models")
+    plt.title(f"Similarity Matrix for {selected_dataset}", fontsize=16)
+    plt.xlabel("Models", fontsize=14)
+    plt.ylabel("Models", fontsize=14)
     plt.xticks(rotation=45, ha='right')
     plt.yticks(rotation=0)
     plt.tight_layout()
@@ -90,7 +94,7 @@ with gr.Blocks(title="LLM Similarity Analyzer") as demo:
     
     clear_btn = gr.Button("Clear Selection")
     clear_btn.click(
-        lambda: [None, None, None],
+        lambda: [[], None, None],
         outputs=[model_dropdown, dataset_dropdown, heatmap]
     )