Zekun Wu commited on
Commit
8abda5f
1 Parent(s): 24180f4
Files changed (1) hide show
  1. pages/2_Injection_Multiple.py +1 -3
pages/2_Injection_Multiple.py CHANGED
@@ -74,9 +74,7 @@ if st.session_state.model_submitted:
74
  # Process data and display results
75
  with st.spinner('Processing data...'):
76
  parameters = {"temperature": st.session_state.temperature, "max_tokens": st.session_state.max_tokens}
77
- df = process_scores(df, st.session_state.num_run, parameters, st.session_state.privilege_label,
78
- st.session_state.protect_label, agent, st.session_state.group_name,
79
- st.session_state.occupation, test_type='multiple')
80
  st.session_state.data_processed = True # Mark as processed
81
 
82
  st.write('Processed Data:', df)
 
74
  # Process data and display results
75
  with st.spinner('Processing data...'):
76
  parameters = {"temperature": st.session_state.temperature, "max_tokens": st.session_state.max_tokens}
77
+ df = process_scores(df, st.session_state.num_run, parameters, st.session_state.privilege_label,st.session_state.protect_label, agent, st.session_state.group_name,st.session_state.occupation, test_type='multiple')
 
 
78
  st.session_state.data_processed = True # Mark as processed
79
 
80
  st.write('Processed Data:', df)