aksell commited on
Commit
beb84f6
·
1 Parent(s): 61419b7

Change default layer and tweak text

Browse files
Files changed (1) hide show
  1. 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=1, min_value=1, max_value=selected_model.layers)
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("Configure parameters", expanded=False):
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: