fix

app.py

@@ -199,17 +199,17 @@ if __name__=='__main__':
             input_ids_dict = {}
             masked_ids_option_1 = {}
             masked_ids_option_2 = {}
-            for sent_id in range(2):
-                option_1_locs[f'sent_{sent_id+1}'], option_2_locs[f'sent_{sent_id+1}'] = separate_options(st.session_state[f'option_locs_{sent_id}'])
-                pron_locs[f'sent_{sent_id+1}'] = st.session_state[f'mask_locs_{sent_id+1}']
-                input_ids_dict[f'sent_{sent_id+1}'] = tokenizer(st.session_state[f'sent_{sent_id+1}']).input_ids
-
-                masked_ids_option_1[f'sent_{sent_id+1}'] = mask_out(input_ids_dict[f'sent_{sent_id+1}'],
-                                                                        pron_locs[f'sent_{sent_id+1}'],
-                                                                        option_1_locs[f'sent_{sent_id+1}'],mask_id)
-                masked_ids_option_2[f'sent_{sent_id+1}'] = mask_out(input_ids_dict[f'sent_{sent_id+1}'],
-                                                                        pron_locs[f'sent_{sent_id+1}'],
-                                                                        option_2_locs[f'sent_{sent_id+1}'],mask_id)
+            for sent_id in [1,2]:
+                option_1_locs[f'sent_{sent_id}'], option_2_locs[f'sent_{sent_id}'] = separate_options(st.session_state[f'option_locs_{sent_id}'])
+                pron_locs[f'sent_{sent_id}'] = st.session_state[f'mask_locs_{sent_id}']
+                input_ids_dict[f'sent_{sent_id}'] = tokenizer(st.session_state[f'sent_{sent_id}']).input_ids
+
+                masked_ids_option_1[f'sent_{sent_id}'] = mask_out(input_ids_dict[f'sent_{sent_id}'],
+                                                                        pron_locs[f'sent_{sent_id}'],
+                                                                        option_1_locs[f'sent_{sent_id}'],mask_id)
+                masked_ids_option_2[f'sent_{sent_id}'] = mask_out(input_ids_dict[f'sent_{sent_id}'],
+                                                                        pron_locs[f'sent_{sent_id}'],
+                                                                        option_2_locs[f'sent_{sent_id}'],mask_id)
 
             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]))