Update app.py

app.py

@@ -41,7 +41,7 @@ def get_model_info(df):
     api = HfApi()
     
     # Initialize new columns for likes and tags
-    df['likes'] = None
+    df[':heart:'] = 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, 'Likes'] = model_info.likes
+            df.loc[index, ':heart:'] = model_info.likes
             df.loc[index, 'Tags'] = ', '.join(model_info.tags)
 
         except (RepositoryNotFoundError, RevisionNotFoundError):
-            df.loc[index, 'Likes'] = -1
+            df.loc[index, ':heart:'] = -1
             df.loc[index, 'Tags'] = ''
 
     return df
@@ -108,9 +108,9 @@ def main():
                 # Toggles
                 col1, col2 = st.columns(2)
                 with col1:
-                    show_phi = st.toggle("Phi (2.8B)", value=True)
+                    show_phi = st.checkbox("Phi (2.8B)", value=True)
                 with col2:
-                    show_mistral = st.toggle("Mistral (7B)", value=True)
+                    show_mistral = st.checkbox("Mistral (7B)", value=True)
 
                 # Apply filters based on toggles
                 if show_phi:
@@ -122,7 +122,7 @@ def main():
                 df = df.sort_values(by='Average', ascending=False)
 
                 # Display the DataFrame
-                st.dataframe(df[['Model'] + score_columns + ['Likes']], use_container_width=True)
+                st.dataframe(df[['Model'] + score_columns + [':heart:']], use_container_width=True)
 
                 # Full-width plot for the first category
                 create_bar_chart(df, score_columns[0])