Update app.py

app.py

@@ -1,6 +1,6 @@
 import streamlit as st
 import pandas as pd
-from huggingface_hub import HfApi, ModelCard
+from huggingface_hub import HfApi, ModelCard, login
 from huggingface_hub.utils import RepositoryNotFoundError, RevisionNotFoundError
 import re
 from io import StringIO
@@ -19,6 +19,9 @@ def cached_model_info(_api, model):
     except (RepositoryNotFoundError, RevisionNotFoundError) as e:
         st.error(f"Error fetching model info for {model}: {str(e)}")
         return None
+    except Exception as e:
+        st.error(f"Unexpected error fetching model info for {model}: {str(e)}")
+        return None
 
 @st.cache_data
 def get_model_info(df):
@@ -96,11 +99,23 @@ def fetch_merge_configs(df):
     except Exception as e:
         st.error(f"Error while fetching merge configs: {str(e)}")
 
+def authenticate_hf():
+    """Authenticate with the Hugging Face API."""
+    token = st.text_input("Enter your Hugging Face API token", type="password")
+    if token:
+        login(token=token)
+        st.success("Authenticated successfully")
+    else:
+        st.warning("You need to enter a Hugging Face API token to access private or gated models")
+
 def main():
     """Main function to set up the Streamlit app and display the leaderboard."""
     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.")
+
+    authenticate_hf()
+
     content = create_yall()
     tab1, tab2 = st.tabs(["🏆 Leaderboard", "📝 About"])