Fix filtering

app.py

@@ -23,7 +23,9 @@ def main():
             placeholder=
             'You can search for titles with regular expressions. e.g. (?<!sur)face',
             max_lines=1)
+        
         case_sensitive = gr.Checkbox(label='Case Sensitive')
+        
         filter_names = gr.CheckboxGroup(choices=[
             'Paper',
             'Code',
@@ -36,10 +38,10 @@ def main():
             'Scientific',
         ]
 
-        # prev: paper_sessions
         data_types = gr.CheckboxGroup(choices=data_type_names,
                                       value=data_type_names,
                                       label='Training Data Type(s)')
+        
         search_button = gr.Button('Search')
 
         number_of_models = gr.Textbox(label='Number of Models Found')

model_list.py

@@ -15,7 +15,7 @@ class ModelList:
 
         self.table_header = '''
             <tr>
-                <td width="40%">Model Name</td>
+                <td width="20%">Model Name</td>
                 <td width="10%">Data Type(s)</td>
                 <td width="10%">Year Published</td>
                 <td width="10%">Paper</td>
@@ -50,9 +50,10 @@ class ModelList:
             rows.append(row)
         self.table['html_table_content'] = rows
 
-    def render(self, search_query: str, case_sensitive: bool,
-               filter_names: list[str],
-               data_types: list[str]) -> tuple[int, str]:
+    def render(self, search_query: str, 
+            case_sensitive: bool,
+            filter_names: list[str],
+            data_types: list[str]) -> tuple[int, str]:
         df = self.table
         if search_query:
             if case_sensitive:
@@ -60,15 +61,14 @@ class ModelList:
             else:
                 df = df[df.name_lowercase.str.contains(search_query.lower())]
         has_paper = 'Paper' in filter_names
-        has_github = 'Github' in filter_names
-        has_model = 'Hub Model' in filter_names or 'Other Weights' in filter_names
+        has_github = 'Code' in filter_names
+        has_model = 'Model Weights' in filter_names
         df = self.filter_table(df, has_paper, has_github, has_model, data_types)
         return len(df), self.to_html(df, self.table_header)
 
     @staticmethod
     def filter_table(df: pd.DataFrame, has_paper: bool, has_github: bool,
-                     has_model: bool,
-                     data_types: list[str]) -> pd.DataFrame:
+                     has_model: bool, data_types: list[str]) -> pd.DataFrame:
         if has_paper:
             df = df[~df.paper.isna()]
         if has_github: