Spaces:
Running
Running
Zekun Wu
commited on
Commit
•
8808aa8
1
Parent(s):
2abe0cf
update
Browse files- util/injection.py +2 -2
util/injection.py
CHANGED
@@ -36,7 +36,7 @@ def create_summary(row, group_name, label, occupation, additional_charateristics
|
|
36 |
|
37 |
summary.append(info)
|
38 |
|
39 |
-
summary.append("\n" + additional_charateristics)
|
40 |
|
41 |
# for column, value in row.items():
|
42 |
# if group_name.lower() not in column.lower():
|
@@ -96,7 +96,7 @@ def process_scores_multiple(df, num_run, parameters, privilege_label, protect_la
|
|
96 |
|
97 |
if additional_charateristics == True:
|
98 |
additional_charateristics = [
|
99 |
-
create_charateristics(
|
100 |
|
101 |
for key, label in zip(['Privilege', 'Protect', 'Neutral'], [privilege_label, protect_label, False]):
|
102 |
prompt_temp = create_summary(row, group_name, label, occupation, additional_charateristics)
|
|
|
36 |
|
37 |
summary.append(info)
|
38 |
|
39 |
+
summary.append("\n" + additional_charateristics) if additional_charateristics else ""
|
40 |
|
41 |
# for column, value in row.items():
|
42 |
# if group_name.lower() not in column.lower():
|
|
|
96 |
|
97 |
if additional_charateristics == True:
|
98 |
additional_charateristics = [
|
99 |
+
create_charateristics(row['resume'], group_name, occupation, agent, parameters)]
|
100 |
|
101 |
for key, label in zip(['Privilege', 'Protect', 'Neutral'], [privilege_label, protect_label, False]):
|
102 |
prompt_temp = create_summary(row, group_name, label, occupation, additional_charateristics)
|