sonic314 commited on
Commit
48d77c5
1 Parent(s): c84d7cc

Change selection order in UI

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -4,20 +4,20 @@ from transformer_ranker import TransformerRanker, prepare_popular_models
4
 
5
  st.title("Choose Your Transformer")
6
 
7
- # 1) Select multiple models from HuggingFace model hub
8
- model_options = ['prajjwal1/bert-tiny', 'google/electra-small-discriminator', 'microsoft/deberta-v3-small',
9
- 'bert-base-uncased', 'bert-base-cased', 'distilbert-base-uncased', 'roberta-base'] # List of models
10
- selected_models = st.multiselect("Select Models", model_options, default=model_options[:1])
11
-
12
- # 2) Select dataset names (from text classification or token classification subcategory)
13
  dataset_options = ['trec', 'conll2003'] # Example datasets; you can expand this list
14
  selected_dataset = st.selectbox("Select Dataset", dataset_options)
15
 
16
- # 3) Select the parameter for dataset downsampling
17
  downsample_values = [0.05, 0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1.0]
18
  downsample_ratio = st.select_slider("Dataset Downsample Ratio", options=downsample_values, value=0.2)
19
 
20
- # 4) Select the parameter for "layer selection" with layermean as the default
 
 
 
 
 
21
  layer_options = ['lastlayer', 'layermean', 'bestlayer']
22
  selected_layer = st.selectbox("Layer Selection", layer_options, index=1)
23
 
 
4
 
5
  st.title("Choose Your Transformer")
6
 
7
+ # 1) Select dataset names (from text classification or token classification subcategory)
 
 
 
 
 
8
  dataset_options = ['trec', 'conll2003'] # Example datasets; you can expand this list
9
  selected_dataset = st.selectbox("Select Dataset", dataset_options)
10
 
11
+ # 2) Select the parameter for dataset downsampling
12
  downsample_values = [0.05, 0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1.0]
13
  downsample_ratio = st.select_slider("Dataset Downsample Ratio", options=downsample_values, value=0.2)
14
 
15
+ # 3) Select multiple models from HuggingFace model hub
16
+ model_options = ['prajjwal1/bert-tiny', 'google/electra-small-discriminator', 'microsoft/deberta-v3-small',
17
+ 'bert-base-uncased', 'bert-base-cased', 'distilbert-base-uncased', 'roberta-base'] # List of models
18
+ selected_models = st.multiselect("Select Models", model_options, default=model_options[:1])
19
+
20
+ # 4) Select the parameter for layer selection with layermean as the default
21
  layer_options = ['lastlayer', 'layermean', 'bestlayer']
22
  selected_layer = st.selectbox("Layer Selection", layer_options, index=1)
23