Update app.py

app.py

@@ -6,9 +6,29 @@ from io import StringIO
 import plotly.graph_objs as go
 from huggingface_hub import HfApi
 from huggingface_hub.utils import RepositoryNotFoundError, RevisionNotFoundError
-
 from yall import create_yall
+from functools import cache
 
+@cache
+def cached_model_info(api, model):
+    try:
+        return api.model_info(repo_id=str(model))
+    except (RepositoryNotFoundError, RevisionNotFoundError):
+        return None
+
+@st.cache
+def get_model_info(df):
+    api = HfApi()
+
+    for index, row in df.iterrows():
+        model_info = cached_model_info(api, row['Model'].strip())
+        if model_info:
+            df.loc[index, 'Likes'] = model_info.likes
+            df.loc[index, 'Tags'] = ', '.join(model_info.tags)
+        else:
+            df.loc[index, 'Likes'] = -1
+            df.loc[index, 'Tags'] = ''
+    return df
 
 
 def convert_markdown_table_to_dataframe(md_content):
@@ -154,7 +174,9 @@ def main():
                     hide_index=True,
                     height=len(df) * 37,
                 )
-
+                selected_models = st.multiselect('Select models to compare', df['Model'].unique())
+                comparison_df = df[df['Model'].isin(selected_models)]
+                st.dataframe(comparison_df)
                 # Add a button to export data to CSV
                 if st.button("Export to CSV"):
                     # Export the DataFrame to CSV