debug

app.py

@@ -237,6 +237,7 @@ if __name__=='__main__':
             for token_ids in [masked_ids_option_1['sent_1'],masked_ids_option_1['sent_2'],masked_ids_option_2['sent_1'],masked_ids_option_2['sent_2']]:
                 st.write(' '.join([tokenizer.decode([token]) for token in token_ids]))
 
+    if st.session_state['page_status'] == 'finish_debug':
             option_1_tokens_1 = np.array(input_ids_dict['sent_1'])[np.array(option_1_locs['sent_1'])+1]
             option_1_tokens_2 = np.array(input_ids_dict['sent_2'])[np.array(option_1_locs['sent_2'])+1]
             option_2_tokens_1 = np.array(input_ids_dict['sent_1'])[np.array(option_2_locs['sent_1'])+1]
@@ -245,7 +246,6 @@ if __name__=='__main__':
             option_1_tokens = option_1_tokens_1
             option_2_tokens = option_2_tokens_1
 
-    if st.session_state['page_status'] == 'finish_debug':
             interventions = [{'lay':[],'qry':[],'key':[],'val':[]} for i in range(num_layers)]
             probs_original = run(interventions,1,model,masked_ids_option_1,masked_ids_option_2,option_1_tokens,option_2_tokens,pron_locs)
             st.write(probs_original)