taka-yamakoshi
commited on
Commit
•
c8e743e
1
Parent(s):
ca1b654
fix
Browse files
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)
|