Zekun Wu commited on
Commit
29abe81
·
1 Parent(s): 3b394b8
Files changed (1) hide show
  1. util/injection.py +1 -1
util/injection.py CHANGED
@@ -99,7 +99,7 @@ def process_scores_multiple(df, num_run, parameters, privilege_label, protect_la
99
 
100
  if additional_charateristics == True:
101
  additional_charateristics = [
102
- create_charateristics(row['resume'], group_name, occupation, agent, parameters)]
103
 
104
  for key, label in zip(['Privilege', 'Protect', 'Neutral'], [privilege_label, protect_label, False]):
105
  prompt_temp = create_summary(row, group_name, label, occupation, additional_charateristics)
 
99
 
100
  if additional_charateristics == True:
101
  additional_charateristics = [
102
+ create_charateristics('\n'.join(summary), group_name, occupation, agent, parameters)]
103
 
104
  for key, label in zip(['Privilege', 'Protect', 'Neutral'], [privilege_label, protect_label, False]):
105
  prompt_temp = create_summary(row, group_name, label, occupation, additional_charateristics)