Update app.py

app.py

@@ -18,6 +18,10 @@ from bigcodebench.data import get_bigcodebench, get_bigcodebench_hash, load_solu
 from bigcodebench.data.utils import CACHE_DIR
 from bigcodebench.eval import PASS, compatible_eval_result, estimate_pass_at_k, untrusted_check
 from bigcodebench.gen.util import trusted_check
+from apscheduler.schedulers.background import BackgroundScheduler
+
+REPO_ID = "bigcode/bigcodebench-evaluator"
+HF_TOKEN = os.environ.get("HF_TOKEN", None)
 
 Result = Tuple[str, List[bool]]
 
@@ -251,6 +255,22 @@ def run_gradio():
     interface.queue(default_concurrency_limit=None)
     interface.launch(show_error=True)
 
+
+def restart_space():
+    logging.info(f"Restarting space with repo ID: {REPO_ID}")
+    try:
+        # Now restart the space
+        API.restart_space(repo_id=REPO_ID, token=HF_TOKEN)
+        logging.info("Space restarted successfully.")
+    except Exception as e:
+        logging.error(f"Failed to restart space: {e}")
+
+
 if __name__ == "__main__":
     run_gradio()
+    
+    scheduler = BackgroundScheduler()
+    scheduler.add_job(restart_space, "interval", hours=1)  # Restart every 1h
+    logging.info("Scheduler initialized to restart space every 1 hour.")
+    scheduler.start()
     # evaluate("complete", "hard", "meta-llama--Llama-3.2-3B-Instruct--bigcodebench-instruct--vllm-0-1.jsonl")