Update

paper_list.py

@@ -1,10 +1,14 @@
 from __future__ import annotations
 
+import numpy as np
 import pandas as pd
+import requests
+from huggingface_hub.hf_api import SpaceInfo
 
 
 class PaperList:
     def __init__(self):
+        self.organization_name = 'ICML2022'
         self.table = pd.read_csv('papers.csv')
         self._preprcess_table()
 
@@ -20,7 +24,30 @@ class PaperList:
                 <td width="4%">HF Datasets</td>
             </tr>'''
 
+    @staticmethod
+    def load_space_info(author: str) -> list[SpaceInfo]:
+        path = 'https://huggingface.co/api/spaces'
+        r = requests.get(path, params={'author': author})
+        d = r.json()
+        return [SpaceInfo(**x) for x in d]
+
+    def add_spaces_to_table(self, organization_name: str,
+                            df: pd.DataFrame) -> pd.DataFrame:
+        spaces = self.load_space_info(organization_name)
+        name2space = {
+            s.id.split('/')[1].lower(): f'https://huggingface.co/spaces/{s.id}'
+            for s in spaces
+        }
+        df['hf_space'] = df.loc[:, ['hf_space', 'github']].apply(
+            lambda x: x[0] if isinstance(x[0], str) else name2space.get(
+                x[1].split('/')[-1].lower()
+                if isinstance(x[1], str) else '', np.nan),
+            axis=1)
+        return df
+
     def _preprcess_table(self) -> None:
+        self.table = self.add_spaces_to_table(self.organization_name,
+                                              self.table)
         self.table['title_lowercase'] = self.table.title.str.lower()
 
         rows = []
@@ -53,7 +80,7 @@ class PaperList:
 
     def render(self, search_query: str, case_sensitive: bool,
                filter_names: list[str]) -> tuple[int, str]:
-        df = self.table
+        df = self.add_spaces_to_table(self.organization_name, self.table)
         if search_query:
             if case_sensitive:
                 df = df[df.title.str.contains(search_query)]

papers.csv

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:630a7034176c44dda64b1eab05568eacc8df8001e8065caedf7aa7e20c4f09bb
-size 335364
+oid sha256:8264ef7763719ee8f8aeebc99926f97e8338eaeb70864242171c4cdb21930ac2
+size 335399