taka-yamakoshi commited on
Commit
c8e743e
1 Parent(s): ca1b654
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -345,7 +345,7 @@ if __name__=='__main__':
345
  if multihead:
346
  show_results(effect_array[:,:,0],masked_ids_option_1['sent_1'],token_id_list,num_layers)
347
  else:
348
- tabs = st.tabs(heads)
349
  for i,tab in enumerate(tabs):
350
  with tab:
351
  show_results(effect_array[:,:,i],masked_ids_option_1['sent_1'],token_id_list,num_layers)
 
345
  if multihead:
346
  show_results(effect_array[:,:,0],masked_ids_option_1['sent_1'],token_id_list,num_layers)
347
  else:
348
+ tabs = st.tabs([str(head_id) for head_id in heads])
349
  for i,tab in enumerate(tabs):
350
  with tab:
351
  show_results(effect_array[:,:,i],masked_ids_option_1['sent_1'],token_id_list,num_layers)