Refresh leaderboard every 20 minutes (#2)

- Restart every 20 mins (650c7423191c22f3b96f8ca397486093f3cdf758)


Co-authored-by: Omar Sanseviero <[email protected]>

app.py

@@ -1,9 +1,16 @@
-from datasets import load_dataset
-from collections import Counter, defaultdict
-import pandas as pd
-from huggingface_hub import list_datasets
 import os
+
 import gradio as gr
+import pandas as pd
+
+from apscheduler.schedulers.background import BackgroundScheduler
+from collections import Counter, defaultdict
+from datasets import load_dataset
+from huggingface_hub import HfApi, list_datasets
+
+api = HfApi(token=os.environ.get("HF_TOKEN", None))
+def restart_space():
+  api.restart_space(repo_id="OpenGenAI/parti-prompts-leaderboard")
 
 parti_prompt_results = []
 ORG = "diffusers-parti-prompts"
@@ -176,6 +183,10 @@ with gr.Blocks() as demo:
     with gr.Row():
         refresh_button = gr.Button("Refresh")
         refresh_button.click(refresh, inputs=[], outputs=[num_submissions, main_dataframe, cat_dataframe, chal_dataframe])   
-    demo.load(refresh, inputs=[], outputs=[num_submissions, main_dataframe, cat_dataframe, chal_dataframe])
+
+# Restart space every 20 minutes
+scheduler = BackgroundScheduler()
+scheduler.add_job(restart_space, 'interval', seconds=3600)
+scheduler.start()
 
 demo.launch()

requirements.txt

@@ -1,2 +1,3 @@
+apscheduler
 datasets
 pandas