Update app.py

app.py

@@ -256,19 +256,24 @@ def run_gradio():
     interface.launch(show_error=True)
 
 
+def preload_gt():
+    scheduler.start(split="complete", subset="full", samples="", check_gt_only=True)
+    scheduler.start(split="complete", subset="hard", samples="", check_gt_only=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)
-        scheduler.start(split="complete", subset="full", samples="", check_gt_only=True)
-        scheduler.start(split="complete", subset="hard", samples="", check_gt_only=True)
+        preload_gt()
         logging.info("Space restarted successfully.")
     except Exception as e:
         logging.error(f"Failed to restart space: {e}")
 
 
 if __name__ == "__main__":
+    preload_gt()
     run_gradio()
     
     scheduler = BackgroundScheduler()