ercanburak commited on
Commit
248985f
1 Parent(s): 18319a3

make ECD - dynamic_6dof default sequence

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -194,12 +194,18 @@ assert len(set(viz_display_names)) == len(viz_display_names), "Viz display names
194
 
195
  col1, col2 = st.columns(2)
196
 
 
 
 
197
  with col1:
198
- selected_dataset_name = st.selectbox('Select dataset:', options=dataset_display_names)
 
199
  selected_dataset = [dataset for dataset in datasets if dataset['display_name'] == selected_dataset_name][0]
200
 
201
  with col2:
202
- selected_sequence = st.selectbox('Select sequence:', options=selected_dataset["sequences"].keys())
 
 
203
 
204
  selected_model_names = st.multiselect('Select methods to compare:', model_display_names)
205
  selected_models = [models[model_display_names.index(model_name)] for model_name in selected_model_names]
 
194
 
195
  col1, col2 = st.columns(2)
196
 
197
+ default_dataset = "ECD"
198
+ default_sequence = "dynamic_6dof"
199
+
200
  with col1:
201
+ default_dataset_index = dataset_display_names.index(default_dataset) if default_dataset in dataset_display_names else 0
202
+ selected_dataset_name = st.selectbox('Select dataset:', options=dataset_display_names, index=default_dataset_index)
203
  selected_dataset = [dataset for dataset in datasets if dataset['display_name'] == selected_dataset_name][0]
204
 
205
  with col2:
206
+ dataset_sequences = list(selected_dataset["sequences"].keys())
207
+ default_sequence_index = dataset_sequences.index(default_sequence) if default_sequence in dataset_sequences else 0
208
+ selected_sequence = st.selectbox('Select sequence:', options=dataset_sequences, index=default_sequence_index)
209
 
210
  selected_model_names = st.multiselect('Select methods to compare:', model_display_names)
211
  selected_models = [models[model_display_names.index(model_name)] for model_name in selected_model_names]