Update app.py

app.py

@@ -2,21 +2,31 @@ import altair as alt
 import gradio as gr
 import pandas as pd
 
+from functools import partial
 from datasets import load_dataset
 
-model_id = "ybelkada/model_cards_correct_tag"
-dataset = load_dataset(model_id, split="train").to_pandas()
+def get_data():
+    model_id = "ybelkada/model_cards_correct_tag"
+    dataset = load_dataset(model_id, split="train").to_pandas()
 
-# Convert dataset to a pandas DataFrame and sort by commit_dates
-df = pd.DataFrame(dataset)
-df["commit_dates"] = pd.to_datetime(df["commit_dates"])  # Convert commit_dates to datetime format
-df = df.sort_values(by="commit_dates")
-melted_df = pd.melt(df, id_vars=['commit_dates'], value_vars=['total_transformers_model', 'missing_library_name'], var_name='type')
+    # Convert dataset to a pandas DataFrame and sort by commit_dates
+    df = pd.DataFrame(dataset)
+    df["commit_dates"] = pd.to_datetime(df["commit_dates"])  # Convert commit_dates to datetime format
+    df = df.sort_values(by="commit_dates")
+    melted_df = pd.melt(df, id_vars=['commit_dates'], value_vars=['total_transformers_model', 'missing_library_name'], var_name='type')
 
-df['ratio'] = (1 - df['missing_library_name'] / df['total_transformers_model']) * 100
-ratio_df = df[['commit_dates', 'ratio']].copy()
+    df['ratio'] = (1 - df['missing_library_name'] / df['total_transformers_model']) * 100
+    ratio_df = df[['commit_dates', 'ratio']].copy()
+    return ratio_df, melted_df
+
+ratio_df, melted_df = get_data()
+
+def make_plot(plot_type, refresh=False):
+    global ratio_df, melted_df
+
+    if refresh:
+        ratio_df, melted_df = get_data()
 
-def make_plot(plot_type):
     if plot_type == "Total models with missing 'transformers' tag":
         highlight = alt.selection(type='single', on='mouseover',
                                 fields=['type'], nearest=True)
@@ -65,6 +75,7 @@ def make_plot(plot_type):
         )
         
         return points + lines
+    
 
 with gr.Blocks() as demo:
     button = gr.Radio(
@@ -72,9 +83,13 @@ with gr.Blocks() as demo:
         choices=["Total models with missing 'transformers' tag", "Proportion of models correctly tagged with 'transformers' tag"], 
         value="Total models with missing 'transformers' tag"
     )
+    refresh_button = gr.Button(value="Fetch latest data")
+
     plot = gr.Plot(label="Plot")
-    button.change(make_plot, inputs=button, outputs=[plot])
+
+    button.change(make_plot, inputs=[button], outputs=[plot])
+    refresh_button.click(partial(make_plot, True), inputs=[button], outputs=[plot])
     demo.load(make_plot, inputs=[button], outputs=[plot])
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()