updates

baseline_static_explanations.py

@@ -100,7 +100,8 @@ def generate_explanations(args):
         style_descirption, a2_embeddings = find_closest_cluster_style(instance['a2_fullText'], interp_space, model_name=model_name)
         json_obj['a2_top_style_feats'] = style_descirption
 
-
+        json_obj['gt_idx'] = instance['gt_idx']
+        
         # Compute pairwise similarity between q_embeddings and all a_embeddings
         # Ensure embeddings are 2D arrays for cosine_similarity
         q_emb_2d  = np.array(q_embeddings).reshape(1, -1)