sasha HF Staff commited on
Commit
1e758e9
·
1 Parent(s): 44e43cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -15,15 +15,15 @@ st.markdown("## Choose the dataset you want to use for the comparison:")
15
  api = HfApi()
16
  datasets = api.list_datasets(filter="task_categories:text-classification", sort = "downloads", direction=-1, limit = 20)
17
 
18
- dset = st.selectbox(datasets)
19
 
20
  st.markdown("## Now select up to 5 models to compare their performance:")
21
 
22
  models = api.list_models(filter="trained_dataset:imdb", sort = "downloads", direction=-1, limit = 20)
23
 
24
- options = st.multiselect(
25
  'Choose your models',
26
- models)
27
 
28
  """
29
  tokenizer1 = AutoTokenizer.from_pretrained("lvwerra/distilbert-imdb")
 
15
  api = HfApi()
16
  datasets = api.list_datasets(filter="task_categories:text-classification", sort = "downloads", direction=-1, limit = 20)
17
 
18
+ dset = st.selectbox(options=datasets)
19
 
20
  st.markdown("## Now select up to 5 models to compare their performance:")
21
 
22
  models = api.list_models(filter="trained_dataset:imdb", sort = "downloads", direction=-1, limit = 20)
23
 
24
+ model = st.multiselect(
25
  'Choose your models',
26
+ options=models)
27
 
28
  """
29
  tokenizer1 = AutoTokenizer.from_pretrained("lvwerra/distilbert-imdb")