Hugging Face's logo

Spaces:
taka-yamakoshi
/
causal-intervention-demo
Runtime error

App Files Community
taka-yamakoshi commited on
Commit
b4bb2e9
·
1 Parent(s): bd707c0

sort locs

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -94,6 +94,8 @@ def annotate_options(sent_id,sent):
94
  show_annotated_sentence(decoded_sent,
95
  option_locs=st.session_state[f'option_locs_{sent_id}'],
96
  mask_locs=st.session_state[f'mask_locs_{sent_id}'])
 
 
97
 
98
  def show_annotated_sentence(sent,option_locs=[],mask_locs=[]):
99
  disp_style = '"font-family:san serif; color:Black; font-size: 20px"'
 
94
  show_annotated_sentence(decoded_sent,
95
  option_locs=st.session_state[f'option_locs_{sent_id}'],
96
  mask_locs=st.session_state[f'mask_locs_{sent_id}'])
97
+ st.session_state[f'option_locs_{sent_id}'] = list(np.sort(st.session_state[f'option_locs_{sent_id}']))
98
+ st.session_state[f'mask_locs_{sent_id}'] = list(np.sort(st.session_state[f'mask_locs_{sent_id}']))
99
 
100
  def show_annotated_sentence(sent,option_locs=[],mask_locs=[]):
101
  disp_style = '"font-family:san serif; color:Black; font-size: 20px"'