Add columns of Hugging Face Models and Datasets

app.py

@@ -21,6 +21,8 @@ def main():
             'arXiv',
             'GitHub',
             'HF Space',
+            'HF Model',
+            'HF Dataset',
         ],
                                         label='With')
         search_box = gr.Textbox(

paper_list.py

@@ -11,12 +11,14 @@ class PaperList:
         self.table_header = '''
             <tr>
                 <td width="50%">Paper</td>
-                <td width="25%">Authors</td>
-                <td width="5%">pdf</td>
-                <td width="5%">Supp</td>
-                <td width="5%">arXiv</td>
-                <td width="5%">GitHub</td>
-                <td width="5%">HF Spaces</td>
+                <td width="22%">Authors</td>
+                <td width="4%">pdf</td>
+                <td width="4%">Supp</td>
+                <td width="4%">arXiv</td>
+                <td width="4%">GitHub</td>
+                <td width="4%">HF Spaces</td>
+                <td width="4%">HF Models</td>
+                <td width="4%">HF Datasets</td>
             </tr>'''
 
     def _preprcess_table(self) -> None:
@@ -34,6 +36,10 @@ class PaperList:
                 row.github, str) else ''
             hf_space = f'<a href="{row.hf_space}" target="_blank">Space</a>' if isinstance(
                 row.hf_space, str) else ''
+            hf_model = f'<a href="{row.hf_model}" target="_blank">Model</a>' if isinstance(
+                row.hf_model, str) else ''
+            hf_dataset = f'<a href="{row.hf_dataset}" target="_blank">Dataset</a>' if isinstance(
+                row.hf_dataset, str) else ''
             row = f'''
                 <tr>
                     <td>{paper}</td>
@@ -43,6 +49,8 @@ class PaperList:
                     <td>{arxiv}</td>
                     <td>{github}</td>
                     <td>{hf_space}</td>
+                    <td>{hf_model}</td>
+                    <td>{hf_dataset}</td>
                 </tr>'''
             rows.append(row)
         self.table['html_table_content'] = rows
@@ -59,12 +67,15 @@ class PaperList:
         has_arxiv = 'arXiv' in filter_names
         has_github = 'GitHub' in filter_names
         has_hf_space = 'HF Space' in filter_names
+        has_hf_model = 'HF Model' in filter_names
+        has_hf_dataset = 'HF Dataset' in filter_names
         df = self.filter_table(df, has_supp, has_arxiv, has_github,
-                               has_hf_space)
+                               has_hf_space, has_hf_model, has_hf_dataset)
         return len(df), self.to_html(df)
 
     def filter_table(self, df: pd.DataFrame, has_supp: bool, has_arxiv: bool,
-                     has_github: bool, has_hf_space: bool) -> pd.DataFrame:
+                     has_github: bool, has_hf_space: bool, has_hf_model: bool,
+                     has_hf_dataset: bool) -> pd.DataFrame:
         if has_supp:
             df = df[~df.supp.isna()]
         if has_arxiv:
@@ -73,6 +84,10 @@ class PaperList:
             df = df[~df.github.isna()]
         if has_hf_space:
             df = df[~df.hf_space.isna()]
+        if has_hf_model:
+            df = df[~df.hf_model.isna()]
+        if has_hf_dataset:
+            df = df[~df.hf_dataset.isna()]
         return df
 
     def to_html(self, df: pd.DataFrame) -> str:

papers.csv

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:fd0652a3aef2488fc06b90484e2c5de90f64f56b12be22d578e1509a84a9aaa9
-size 1140089
+oid sha256:468cc57c8da9a9f2eac6d9363f7c19221e1f03fe72f41ca0df19f0029185ffb6
+size 1144297