debug

app.py

@@ -209,8 +209,10 @@ if __name__=='__main__':
                                                                         pron_locs[f'sent_{sent_id}'],
                                                                         option_2_locs[f'sent_{sent_id}'],mask_id)
 
+            st.write(option_1_locs)
+            st.write(option_2_locs)
             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 toke in token_ids]))
+                st.write(' '.join([tokenizer.decode([token]) for token in token_ids]))
 
     if st.session_state['page_status'] == 'finish_debug':
             for layer_id in range(num_layers):