update

generation.py

@@ -52,7 +52,8 @@ def process_scores(df, num_run,parameters,privilege_label,protect_label,agent,gr
         for index, row in tqdm(df.iterrows(), total=len(df), desc="Processing entries", unit="entry"):
             for key, label in zip(['Privilege', 'Protect', 'Neutral'], [privilege_label, protect_label, None]):
                 prompt_temp = create_summary(row,group_name,label)
-                print(f"Run {run + 1} - Entry {index + 1} - {key}:\n{prompt_temp}")
+                # print(f"Run {run + 1} - Entry {index + 1} - {key}:\n{prompt_temp}")
+                # print("=============================================================")
                 result = invoke_retry(prompt_temp,agent,parameters)
                 scores[key][index].append(result)