[Fix] sim check for gts values

app.py

@@ -43,7 +43,7 @@ def create_heatmap(selected_models, selected_dataset, selected_metric):
     )
     
     # Customize plot
-    plt.title(f"{selected_metric} Similarities for {selected_dataset}", fontsize=16)
+    plt.title(f"{selected_metric} for {selected_dataset}", fontsize=16)
     plt.xlabel("Models", fontsize=14)
     plt.ylabel("Models", fontsize=14)
     plt.xticks(rotation=45, ha='right')

src/similarity.py

@@ -27,7 +27,7 @@ def compute_similarity(metric: Metrics, probs_a: list[np.array], gt_a: list[int]
     output_b = []
     gt = []
     for i in range(len(probs_a)):
-        if gt_a == gt_b:
+        if gt_a[i] == gt_b[i]:
             output_a.append(probs_a[i])
             output_b.append(probs_b[i])
             gt.append(gt_a[i])