Spaces:
Running
Running
Zekun Wu
commited on
Commit
·
dee7dcf
1
Parent(s):
29abe81
update
Browse files- util/injection.py +4 -3
util/injection.py
CHANGED
@@ -36,7 +36,7 @@ def create_summary(row, group_name, label, occupation, additional_charateristics
|
|
36 |
|
37 |
|
38 |
|
39 |
-
summary.append("\n"
|
40 |
|
41 |
# for column, value in row.items():
|
42 |
# if group_name.lower() not in column.lower():
|
@@ -98,8 +98,9 @@ def process_scores_multiple(df, num_run, parameters, privilege_label, protect_la
|
|
98 |
summary.append(f"{readable_name}: {value};")
|
99 |
|
100 |
if additional_charateristics == True:
|
101 |
-
additional_charateristics =
|
102 |
-
|
|
|
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)
|
|
|
36 |
|
37 |
|
38 |
|
39 |
+
summary.append(f"\n{additional_charateristics}")
|
40 |
|
41 |
# for column, value in row.items():
|
42 |
# if group_name.lower() not in column.lower():
|
|
|
98 |
summary.append(f"{readable_name}: {value};")
|
99 |
|
100 |
if additional_charateristics == True:
|
101 |
+
additional_charateristics = create_charateristics('\n'.join(summary), group_name, occupation, agent, parameters)
|
102 |
+
else:
|
103 |
+
additional_charateristics = ""
|
104 |
|
105 |
for key, label in zip(['Privilege', 'Protect', 'Neutral'], [privilege_label, protect_label, False]):
|
106 |
prompt_temp = create_summary(row, group_name, label, occupation, additional_charateristics)
|