Update app.py

app.py

@@ -31,15 +31,22 @@ def main():
             'Model Weights',
         ], label='Filter')
 
-        # data_type_names = [
-        #     'Biomedical',
-        #     'Clinical',
-        #     'Scientific',
-        # ]
-
-        # data_types = gr.CheckboxGroup(choices=data_type_names,
-        #                               value=data_type_names,
-        #                               label='Training Data Type(s)')
+        data_type_names = [
+            'DNA', 'scRNA', 'protein language model', 'protein folding', 'protein generation', 'antibody folding', 'molecules',
+            'protein-to-ligand', 'reaction-to-enzyme', 'enzyme generation', 
+        ]
+
+        data_types = gr.CheckboxGroup(choices=data_type_names,
+                                      value=data_type_names,
+                                      label='Type')
+
+        model_type_names = [
+            'GPT2', 'GPT-Neo', 'GPT-NeoX', 'ESM', 'BERT', 'RoBERTa', 'BART', 'T5', 'MPNN', 'diffusion', 'custom model'
+        ]
+
+        model_types = gr.CheckboxGroup(choices=model_type_names,
+                                      value=model_type_names,
+                                      label='Base Model')
         
         search_button = gr.Button('Search')
 
@@ -54,7 +61,8 @@ def main():
                       search_box,
                       case_sensitive,
                       filter_names,
-                      #data_types,
+                      data_types,
+                      model_types
                   ],
                   outputs=[
                       number_of_models,
@@ -65,7 +73,8 @@ def main():
                               search_box,
                               case_sensitive,
                               filter_names,
-                              #data_types,
+                              data_types,
+                              model_types
                           ],
                           outputs=[
                               number_of_models,
@@ -77,7 +86,8 @@ def main():
                                 search_box,
                                 case_sensitive,
                                 filter_names,
-                                #data_types,
+                                data_types,
+                                model_types
                             ],
                             outputs=[
                                 number_of_models,