Update app.py

app.py

@@ -2,9 +2,7 @@ import streamlit as st
 import pandas as pd
 from huggingface_hub import HfApi
 from huggingface_hub.utils import RepositoryNotFoundError, RevisionNotFoundError
-from itertools import combinations
 import re
-from functools import cache
 from io import StringIO
 from yall import create_yall
 import plotly.graph_objs as go
@@ -14,7 +12,7 @@ def calculate_pages(df, items_per_page):
     return -(-len(df) // items_per_page)  # Equivalent to math.ceil(len(df) / items_per_page)
 
 @st.cache_data
-def cached_model_info(_api, model):
+def cached_model_info(api, model):
     try:
         return api.model_info(repo_id=str(model))
     except (RepositoryNotFoundError, RevisionNotFoundError):