github stars

df/PaperCentral.py

@@ -70,6 +70,7 @@ class PaperCentral:
         'upvotes',
         'num_comments',
         'github',
+        'github_stars',
         'conference_name',
         'id',
         'type',
@@ -82,8 +83,8 @@ class PaperCentral:
         'date': 'str',
         'arxiv_id': 'markdown',
         'paper_page': 'markdown',
-        'upvotes': 'str',
-        'num_comments': 'str',
+        'upvotes': 'number',
+        'num_comments': 'number',
         'num_models': 'markdown',
         'num_datasets': 'markdown',
         'num_spaces': 'markdown',
@@ -94,6 +95,7 @@ class PaperCentral:
         'id': 'str',
         'type': 'str',
         'authors': 'str',
+        'github_stars': 'number',
     }
 
     # Mapping for renaming columns for display purposes
@@ -101,7 +103,10 @@ class PaperCentral:
         'num_models': 'models',
         'num_spaces': 'spaces',
         'num_datasets': 'datasets',
-        'conference_name': 'venue',
+        'github': 'GitHub',
+        'github_stars': 'GitHub⭐',
+        'num_comments': '💬',
+        'upvotes': '👍',
     }
 
     def __init__(self):
@@ -149,7 +154,7 @@ class PaperCentral:
         paper_central_df: pd.DataFrame = load_and_process(DATASET_PAPER_CENTRAL)[
             ['arxiv_id', 'categories', 'primary_category', 'date', 'upvotes', 'num_comments', 'github', 'num_models',
              'num_datasets', 'num_spaces', 'id', 'proceedings', 'type',
-             'conference_name', 'title', 'paper_page', 'authors']
+             'conference_name', 'title', 'paper_page', 'authors', 'github_stars']
         ]
 
         return paper_central_df
@@ -404,6 +409,7 @@ class PaperCentral:
             if "github" in hf_options:
                 if 'github' not in columns_to_show:
                     columns_to_show.append('github')
+                    columns_to_show.append('github_stars')
                 filtered_df = filtered_df[(filtered_df['github'] != "") & (filtered_df['github'].notnull())]
 
         # Apply conference filtering