add org rankings

README.md

@@ -1,6 +1,6 @@
 ---
 title: Base Model Explorer
-emoji: 🏃
+emoji: 🧭 
 colorFrom: green
 colorTo: purple
 sdk: gradio

app.py

@@ -37,6 +37,14 @@ base_models = [
 df = pd.DataFrame(
     pd.DataFrame({"base_model": base_models}).value_counts()
 ).reset_index()
+df_with_org = df.copy(deep=True)
+
+def parse_org(hub_id):
+    parts = hub_id.split('/')
+    return parts[0] if len(parts) == 2 else None
+
+df_with_org["org"] = df_with_org["base_model"].apply(parse_org)
+df_with_org = df_with_org.dropna(subset=['org'])
 
 grouped_by_base_model = groupby(
     lambda x: x.cardData.get("base_model"), models_with_base_model_info
@@ -61,12 +69,26 @@ def return_models_for_base_model(base_model):
 
 
 with gr.Blocks() as demo:
-    gr.Markdown("### Models with base model info")
+    gr.Markdown("# Base model explorer")
+    gr.Markdown(
+        """When sharing models to the Hub it is possible to specify a base model in the model card i.e. that your model is a fine-tuned version of [bert/base-cased](https://huggingface.co/bert-base-cased). 
+                This Space allows you to find children models for a given base model and view the popularity of models for fine-tuning."""
+    )
     gr.Markdown(summary)
     gr.Markdown("### Find all models trained from a base model")
     base_model = gr.Dropdown(all_base_models, label="Base Model")
     results = gr.Markdown()
     base_model.change(return_models_for_base_model, base_model, results)
+    with gr.Accordion("Base model popularity ranking"):
+        gr.DataFrame(df.head(50))
+    with gr.Accordion("Base model popularity ranking by organisation"):
+        gr.DataFrame(
+            pd.DataFrame(
+            df_with_org.groupby("org")["count"]
+            .sum()
+            .sort_values(ascending=False)
+            .head(50)
+        ).reset_index()).sort_values("count", ascending=False)
+
 
-    # gr.DataFrame(df)
 demo.launch()