Hide min attention slider in expander
Browse filesPeople love a slider. But this is not really the most interesting
first thing to play with, as for most higher attention thresholds
you'll see very few attention pairs.
- hexviz/app.py +3 -1
hexviz/app.py
CHANGED
@@ -33,7 +33,9 @@ with right:
|
|
33 |
head_one = st.number_input("Head", value=1, min_value=1, max_value=selected_model.heads)
|
34 |
head = head_one - 1
|
35 |
|
36 |
-
|
|
|
|
|
37 |
|
38 |
attention_pairs = get_attention_pairs(pdb_id, layer, head, min_attn, model_type=selected_model.name)
|
39 |
|
|
|
33 |
head_one = st.number_input("Head", value=1, min_value=1, max_value=selected_model.heads)
|
34 |
head = head_one - 1
|
35 |
|
36 |
+
|
37 |
+
with st.expander("Configue parameters", expanded=False):
|
38 |
+
min_attn = st.slider("Minimum attention", min_value=0.0, max_value=0.4, value=0.1)
|
39 |
|
40 |
attention_pairs = get_attention_pairs(pdb_id, layer, head, min_attn, model_type=selected_model.name)
|
41 |
|