Update model_list.py

model_list.py

@@ -17,6 +17,7 @@ class ModelList:
             <tr>
                 <td width="20%">Model Name</td>
                 <td width="10%">Type</td>
+                <td width="10%">Year</td>
                 <td width="10%">Paper</td>
                 <td width="10%">Code on Github</td>
                 <td width="10%">Weights on 🤗</td>
@@ -37,13 +38,16 @@ class ModelList:
             other_model = f'<a href="{row.other}" target="_blank">Other Weights</a>' if isinstance(
                 row.other, str) else ''
             data_type = f'{row.data_type}' if isinstance(
-                row.paper, str) else ''
+                row.data_type, str) else ''
             base_model = f'{row.base_model}' if isinstance(
-                row.paper, str) else ''
+                row.base_model, str) else ''
+            year = f'{row.year}' if isinstance(
+                row.year, str) else ''
             row = f'''
                 <tr>
                     <td>{row.name}</td>
                     <td>{data_type}</td>
+                    <td>{year}</td>
                     <td>{paper}</td>
                     <td>{github}</td>
                     <td>{hf_model}</td>
@@ -56,6 +60,7 @@ class ModelList:
             case_sensitive: bool,
             filter_names: list[str],
             data_types: list[str],
+            years: list[str],
             #model_types: list[str]
               ) -> tuple[int, str]:
         df = self.table
@@ -68,6 +73,7 @@ class ModelList:
         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)
+        df = self.filter_table(df, has_paper, has_github, has_model, data_types, years)
         #df = self.filter_table(df, has_paper, has_github, has_model, data_types, model_types)
         return len(df), self.to_html(df, self.table_header)
 
@@ -75,6 +81,7 @@ class ModelList:
     def filter_table(df: pd.DataFrame, has_paper: bool, has_github: bool,
                      has_model: bool,
                      data_types: list[str],
+                     years: list[str],
                      #model_types: list[str]
                     ) -> pd.DataFrame:
         if has_paper:
@@ -85,6 +92,7 @@ class ModelList:
             df = df[~df.hub.isna() | ~df.other.isna()]
         df = df[df.data_type.isin(set(data_types))]
         #df = df[df.base_model.isin(set(model_types))]
+        df = df[df.base_model.isin(set(years))]
         return df
 
     @staticmethod