add restart

app.py

@@ -8,6 +8,7 @@ import uuid
 import glob
 import shutil
 from pathlib import Path
+from huggingface_hub import HfApi
 from apscheduler.schedulers.background import BackgroundScheduler
 
 default_command = "bigcodebench.evaluate"
@@ -182,5 +183,14 @@ with gr.Blocks() as demo:
                  inputs=[command_output, jsonl_file, subset, split],
                  outputs=[log_output, download_btn])
 
+REPO_ID = "bigcode/bigcodebench-evaluator"
+HF_TOKEN = os.environ.get("HF_TOKEN", None)
+API = HfApi(token=HF_TOKEN)
+
+def restart_space():
+    API.restart_space(repo_id=REPO_ID, token=HF_TOKEN)
+
 demo.queue(max_size=300).launch(share=True, server_name="0.0.0.0", server_port=7860)
-scheduler = BackgroundScheduler()
+scheduler = BackgroundScheduler()
+scheduler.add_job(restart_space, "interval", hours=3) # restarted every 3h as backup in case automatic updates are not working
+scheduler.start()