Update app.py

app.py

@@ -42,7 +42,7 @@ def get_model_info(df):
     
     # Initialize new columns for likes and tags
     df[':heart:'] = None
-    df['tags'] = None
+    df['Tags'] = None
 
     # Iterate through DataFrame rows
     for index, row in df.iterrows():
@@ -103,33 +103,32 @@ def main():
                     # Corrected use of pd.to_numeric
                     full_df[col] = pd.to_numeric(full_df[col].str.strip(), errors='coerce')
                 full_df = get_model_info(full_df)
-                full_df['tags'] = full_df['tags'].fillna('')
+                full_df['Tags'] = full_df['Tags'].fillna('')
                 df = pd.DataFrame(columns=full_df.columns)
                 
-                # # Toggles
-                # col1, col2 = st.columns(2)
-                # with col1:
-                #     show_phi = st.checkbox("Phi (2.8B)", value=True)
-                # with col2:
-                #     show_mistral = st.checkbox("Mistral (7B)", value=True)
-
-                # dfs_to_concat = []
+                # Toggles
+                col1, col2 = st.columns(2)
+                with col1:
+                    show_phi = st.checkbox("Phi (2.8B)", value=True)
+                with col2:
+                    show_mistral = st.checkbox("Mistral (7B)", value=True)
+
+                dfs_to_concat = []
                 
-                # if show_phi:
-                #     dfs_to_concat.append(full_df[full_df['tags'].str.lower().str.contains('phi-msft')])
-                # if show_mistral:
-                #     dfs_to_concat.append(full_df[full_df['tags'].str.lower().str.contains('mistral')])
+                if show_phi:
+                    dfs_to_concat.append(full_df[full_df['Tags'].str.lower().str.contains('phi-msft')])
+                if show_mistral:
+                    dfs_to_concat.append(full_df[full_df['Tags'].str.lower().str.contains('mistral')])
                 
-                # # Concatenate the DataFrames
-                # if dfs_to_concat:
-                #     df = pd.concat(dfs_to_concat, ignore_index=True)
+                # Concatenate the DataFrames
+                if dfs_to_concat:
+                    df = pd.concat(dfs_to_concat, ignore_index=True)
                 
-                # # Sort values
-                # df = df.sort_values(by='Average', ascending=False)
+                # Sort values
+                df = df.sort_values(by='Average', ascending=False)
                 
-                # # Display the DataFrame
-                # st.dataframe(df[['Model'] + score_columns + ['Likes']], use_container_width=True)
-                st.dataframe(full_df, use_container_width=True)
+                # Display the DataFrame
+                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])