build

app.py

@@ -258,13 +258,19 @@ if __name__=='__main__':
                                     index=['Sentence 1','Sentence 2'])
             st.dataframe(df.style.highlight_max(axis=1))
 
+            compare_1 = np.array(masked_ids_option_1['sent_1'])!=np.array(masked_ids_option_1['sent_2'])
+            compare_2 = np.array(masked_ids_option_2['sent_1'])!=np.array(masked_ids_option_2['sent_2'])
+            st.write(compare_1)
+            st.write(compare_2)
+
             multihead = True
-            effect_array = []
             assert np.all(np.array(pron_locs['sent_1'])==np.array(pron_locs['sent_2']))
             assert np.all(np.array(option_1_locs['sent_1'])==np.array(option_1_locs['sent_2']))
             assert np.all(np.array(option_2_locs['sent_1'])==np.array(option_2_locs['sent_2']))
             token_id_list = pron_locs['sent_1'] + option_1_locs['sent_1'] + option_2_locs['sent_1']
             st.write(token_id_list)
+
+            effect_array = []
             for token_id in token_id_list:
                 token_id += 1
                 effect_list = []