Code cleanup and added comparison

app.py

@@ -1,3 +1,4 @@
+# Importing necessary libraries
 import re
 import streamlit as st
 import requests
@@ -9,6 +10,9 @@ from huggingface_hub.utils import RepositoryNotFoundError, RevisionNotFoundError
 from yall import create_yall
 from functools import cache
 
+
+
+# Function to get model info from Hugging Face API using caching
 @cache
 def cached_model_info(api, model):
     try:
@@ -16,6 +20,7 @@ def cached_model_info(api, model):
     except (RepositoryNotFoundError, RevisionNotFoundError):
         return None
 
+# Function to get model info from DataFrame and update it with likes and tags
 @st.cache
 def get_model_info(df):
     api = HfApi()
@@ -30,7 +35,7 @@ def get_model_info(df):
             df.loc[index, 'Tags'] = ''
     return df
 
-
+# Function to convert markdown table to DataFrame and extract Hugging Face URLs
 def convert_markdown_table_to_dataframe(md_content):
     """
     Converts markdown table to Pandas DataFrame, handling special characters and links,
@@ -79,8 +84,7 @@ def get_model_info(df):
 
     return df
 
-
-
+# Function to create bar chart for a given category
 def create_bar_chart(df, category):
     """Create and display a bar chart for a given category."""
     st.write(f"### {category} Scores")
@@ -104,16 +108,15 @@ def create_bar_chart(df, category):
     # Adjust the height of the chart based on the number of rows in the DataFrame
     st.plotly_chart(fig, use_container_width=True, height=len(df) * 35)
 
-# Example usage:
-# create_bar_chart(your_dataframe, 'Your_Category')
-
-
+# Main function to run the Streamlit app
 def main():
+    # Set page configuration and title
     st.set_page_config(page_title="YALL - Yet Another LLM Leaderboard", layout="wide")
 
     st.title("🏆 YALL - Yet Another LLM Leaderboard")
     st.markdown("Leaderboard made with 🧐 [LLM AutoEval](https://github.com/mlabonne/llm-autoeval) using [Nous](https://huggingface.co/NousResearch) benchmark suite.")
 
+    # Create tabs for leaderboard and about section
     content = create_yall()
     tab1, tab2 = st.tabs(["🏆 Leaderboard", "📝 About"])
 
@@ -125,9 +128,11 @@ def main():
 
                 # Display dataframe
                 full_df = convert_markdown_table_to_dataframe(content)
+
                 for col in score_columns:
                     # 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('')
                 df = pd.DataFrame(columns=full_df.columns)
@@ -241,5 +246,6 @@ def main():
             A special thanks to [gblazex](https://huggingface.co/gblazex) for providing many evaluations.
         ''')
 
+# Run the main function if this script is run directly
 if __name__ == "__main__":
     main()