Change selection order in UI
Browse files
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
|
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 |
-
#
|
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 |
-
#
|
|
|
|
|
|
|
|
|
|
|
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 |
|