katielink commited on
Commit
0a4ba8c
·
1 Parent(s): b64ee01

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -9
app.py CHANGED
@@ -40,13 +40,13 @@ def main():
40
  value=data_type_names,
41
  label='Type')
42
 
43
- model_type_names = [
44
- 'GPT2', 'GPT-Neo', 'GPT-NeoX', 'ESM', 'BERT', 'RoBERTa', 'BART', 'T5', 'MPNN', 'diffusion', 'custom model'
45
- ]
46
 
47
- model_types = gr.CheckboxGroup(choices=model_type_names,
48
- value=model_type_names,
49
- label='Base Model')
50
 
51
  search_button = gr.Button('Search')
52
 
@@ -62,7 +62,7 @@ def main():
62
  case_sensitive,
63
  filter_names,
64
  data_types,
65
- model_types
66
  ],
67
  outputs=[
68
  number_of_models,
@@ -74,7 +74,7 @@ def main():
74
  case_sensitive,
75
  filter_names,
76
  data_types,
77
- model_types
78
  ],
79
  outputs=[
80
  number_of_models,
@@ -87,7 +87,7 @@ def main():
87
  case_sensitive,
88
  filter_names,
89
  data_types,
90
- model_types
91
  ],
92
  outputs=[
93
  number_of_models,
 
40
  value=data_type_names,
41
  label='Type')
42
 
43
+ # model_type_names = [
44
+ # 'GPT2', 'GPT-Neo', 'GPT-NeoX', 'ESM', 'BERT', 'RoBERTa', 'BART', 'T5', 'MPNN', 'diffusion', 'custom model'
45
+ # ]
46
 
47
+ # model_types = gr.CheckboxGroup(choices=model_type_names,
48
+ # value=model_type_names,
49
+ # label='Base Model')
50
 
51
  search_button = gr.Button('Search')
52
 
 
62
  case_sensitive,
63
  filter_names,
64
  data_types,
65
+ #model_types
66
  ],
67
  outputs=[
68
  number_of_models,
 
74
  case_sensitive,
75
  filter_names,
76
  data_types,
77
+ #model_types
78
  ],
79
  outputs=[
80
  number_of_models,
 
87
  case_sensitive,
88
  filter_names,
89
  data_types,
90
+ #model_types
91
  ],
92
  outputs=[
93
  number_of_models,