Update app.py

app.py

@@ -41,7 +41,7 @@ def get_model_info(df):
     api = HfApi()
     
     # Initialize new columns for likes and tags
-    df['♥'] = None
+    df['Likes'] = None
     df['Tags'] = None
 
     # Iterate through DataFrame rows
@@ -49,11 +49,11 @@ def get_model_info(df):
         model = row['Model'].strip()
         try:
             model_info = api.model_info(repo_id=str(model))
-            df.loc[index, '♥'] = model_info.likes
+            df.loc[index, 'Likes'] = model_info.likes
             df.loc[index, 'Tags'] = ', '.join(model_info.tags)
 
         except (RepositoryNotFoundError, RevisionNotFoundError):
-            df.loc[index, '♥'] = -1
+            df.loc[index, 'Likes'] = -1
             df.loc[index, 'Tags'] = ''
 
     return df
@@ -122,13 +122,24 @@ def main():
                 
                 # Concatenate the DataFrames
                 if dfs_to_concat:
-                    df = pd.concat(dfs_to_concat, ignore_index=True)
+                    df = pd.concat(
+                        dfs_to_concat,
+                        column_config={
+                            "Likes": st.column_config.NumberColumn(
+                                "Likes",
+                                help="Number of likes on Hugging Face",
+                                format="%d ❤️",
+                            ),
+                            "URL": st.column_config.LinkColumn("App URL"),
+                        },
+                        hide_index=True,
+                    )
                 
                 # Sort values
                 df = df.sort_values(by='Average', ascending=False)
                 
                 # Display the DataFrame
-                st.dataframe(df[['Model'] + score_columns + ['♥']], use_container_width=True)
+                st.dataframe(df[['Model'] + score_columns + ['Likes', 'URL']], use_container_width=True)
 
                 # Full-width plot for the first category
                 create_bar_chart(df, score_columns[0])