add restart_space schedule

app.py

@@ -1,7 +1,8 @@
 import os
 
 import gradio as gr
-from huggingface_hub import HfFileSystem
+from apscheduler.schedulers.background import BackgroundScheduler
+from huggingface_hub import HfFileSystem, HfApi
 
 from src.assets.text_content import INTRODUCTION_TEXT, TITLE
 
@@ -10,8 +11,12 @@ from src.load_from_hub import load_from_hub
 RESULTS_REPO = os.environ.get("RESULTS_REPO", "datasets/vlsp-2023-vllm/vllms-leaderboard")
 HIDDEN_TOKEN = os.environ.get("HIDDEN_TOKEN")
 
+api = HfApi(token=HIDDEN_TOKEN)
 fs = HfFileSystem(token=HIDDEN_TOKEN)
 
+def restart_space():
+    api.restart_space(repo_id="HuggingFaceH4/open_llm_leaderboard", token=HIDDEN_TOKEN)
+
 leaderboard_df = load_from_hub(fs, RESULTS_REPO)
 
 demo = gr.Blocks()
@@ -34,4 +39,7 @@ with demo:
                         value=leaderboard_df,
                     )
 
+scheduler = BackgroundScheduler()
+scheduler.add_job(restart_space, "interval", seconds=1800)
+scheduler.start()
 demo.launch()

requirements.txt

@@ -0,0 +1 @@
+APScheduler==3.10.1