Spaces:
Sleeping
Sleeping
Change default layer and tweak text
Browse files- hexviz/app.py +5 -4
hexviz/app.py
CHANGED
@@ -21,8 +21,6 @@ models = [
|
|
21 |
selected_model_name = st.selectbox("Select a model", [model.name.value for model in models], index=0)
|
22 |
selected_model = next((model for model in models if model.name.value == selected_model_name), None)
|
23 |
|
24 |
-
st.sidebar.title("Settings")
|
25 |
-
|
26 |
pdb_id = st.sidebar.text_input(
|
27 |
label="PDB ID",
|
28 |
value="4RW0",
|
@@ -39,15 +37,18 @@ label_resi = st.sidebar.checkbox(label="Label Residues", value=True)
|
|
39 |
|
40 |
left, right = st.columns(2)
|
41 |
with left:
|
42 |
-
layer_one = st.number_input("Layer", value=
|
43 |
layer = layer_one - 1
|
44 |
with right:
|
45 |
head_one = st.number_input("Head", value=1, min_value=1, max_value=selected_model.heads)
|
46 |
head = head_one - 1
|
47 |
|
48 |
|
49 |
-
with st.expander("
|
50 |
min_attn = st.slider("Minimum attention", min_value=0.0, max_value=0.4, value=0.1)
|
|
|
|
|
|
|
51 |
try:
|
52 |
ec_class = structure.header["compound"]["1"]["ec"]
|
53 |
except KeyError:
|
|
|
21 |
selected_model_name = st.selectbox("Select a model", [model.name.value for model in models], index=0)
|
22 |
selected_model = next((model for model in models if model.name.value == selected_model_name), None)
|
23 |
|
|
|
|
|
24 |
pdb_id = st.sidebar.text_input(
|
25 |
label="PDB ID",
|
26 |
value="4RW0",
|
|
|
37 |
|
38 |
left, right = st.columns(2)
|
39 |
with left:
|
40 |
+
layer_one = st.number_input("Layer", value=10, min_value=1, max_value=selected_model.layers)
|
41 |
layer = layer_one - 1
|
42 |
with right:
|
43 |
head_one = st.number_input("Head", value=1, min_value=1, max_value=selected_model.heads)
|
44 |
head = head_one - 1
|
45 |
|
46 |
|
47 |
+
with st.expander("Attention parameters", expanded=False):
|
48 |
min_attn = st.slider("Minimum attention", min_value=0.0, max_value=0.4, value=0.1)
|
49 |
+
|
50 |
+
# TODO add avg or max attention as params
|
51 |
+
|
52 |
try:
|
53 |
ec_class = structure.header["compound"]["1"]["ec"]
|
54 |
except KeyError:
|