add automated scheduler for daily restarts

app.py

@@ -3,6 +3,9 @@ import pickle
 import pandas as pd
 import gradio as gr
 import plotly.express as px
+from datetime import datetime
+from huggingface_hub import HfApi
+from apscheduler.schedulers.background import BackgroundScheduler
 
 from utils import (
     KEY_TO_CATEGORY_NAME,
@@ -13,6 +16,21 @@ from utils import (
     format_data,
 )
 
+###################
+### Initialize scheduler
+###################
+
+
+def restart_space():
+    HfApi().restart_space(repo_id="andrewrreed/closed-vs-open-arena-elo")
+    print(f"Space restarted on {datetime.now()}")
+
+
+# restart the space every day at 9am
+scheduler = BackgroundScheduler()
+scheduler.add_job(restart_space, "cron", day_of_week="mon-sun", hour=9, minute=0)
+scheduler.start()
+
 ###################
 ### Load Data
 ###################

requirements.txt

@@ -2,4 +2,5 @@ huggingface_hub
 pandas
 plotly
 gradio
-statsmodels
+statsmodels
+apscheduler