Add columns

app.py

@@ -35,6 +35,8 @@ DEFAULT_COLUMNS = [
     "Title",
     "Type",
     "Paper page",
+    "👍",
+    "💬",
     "OpenReview",
     "GitHub",
     "Spaces",

papers.py

@@ -24,6 +24,8 @@ class PaperList:
         ["Authors", "str"],
         ["Type", "str"],
         ["Paper page", "markdown"],
+        ["👍", "number"],
+        ["💬", "number"],
         ["OpenReview", "markdown"],
         ["GitHub", "markdown"],
         ["Spaces", "markdown"],
@@ -40,11 +42,12 @@ class PaperList:
     def get_df() -> pd.DataFrame:
         df = pd.merge(
             left=datasets.load_dataset("ICLR2024/ICLR2024-papers", split="train").to_pandas(),
-            right=datasets.load_dataset("ICLR2024/ICLR2024-num-claimed-papers", split="train").to_pandas(),
+            right=datasets.load_dataset("ICLR2024/ICLR2024-paper-stats", split="train").to_pandas(),
             on="id",
             how="left",
         )
-        df[["n_authors", "n_linked_authors"]] = df[["n_authors", "n_linked_authors"]].fillna(-1).astype(int)
+        keys = ["n_authors", "n_linked_authors", "upvotes", "num_comments"]
+        df[keys] = df[keys].fillna(-1).astype(int)
         df["paper_page"] = df["arxiv_id"].apply(
             lambda arxiv_id: f"https://huggingface.co/papers/{arxiv_id}" if arxiv_id else ""
         )
@@ -62,12 +65,16 @@ class PaperList:
             n_linked_authors = "" if row.n_linked_authors == -1 else row.n_linked_authors
             n_authors = "" if row.n_authors == -1 else row.n_authors
             claimed_paper = "" if n_linked_authors == "" else f"{n_linked_authors}/{n_authors} {author_linked}"
+            upvotes = "" if row.upvotes == -1 else row.upvotes
+            num_comments = "" if row.num_comments == -1 else row.num_comments
 
             new_row = {
                 "Title": row["title"],
                 "Authors": ", ".join(row["authors"]),
                 "Type": row["type"],
                 "Paper page": PaperList.create_link(row["arxiv_id"], row["paper_page"]),
+                "👍": upvotes,
+                "💬": num_comments,
                 "OpenReview": PaperList.create_link("OpenReview", row["OpenReview"]),
                 "GitHub": "\n".join([PaperList.create_link("GitHub", url) for url in row["GitHub"]]),
                 "Spaces": "\n".join(