Upload restart_space.py

restart_space.py

@@ -0,0 +1,24 @@
+import os
+import time
+
+import requests
+import schedule
+
+HF_TOKEN = os.getenv("RESTART_SPACE_TOKEN")
+url = "https://huggingface.co/api/spaces/dedoc/README/restart"
+
+
+def restart_space():
+    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
+    response = requests.post(url, headers=headers)
+    if response.status_code == 200:
+        print("Space restarted successfully!")
+    else:
+        print(f"Failed to restart: {response.status_code}")
+
+
+if __name__ == "__main__":
+    schedule.every().hour.do(restart_space)
+    while True:
+        schedule.run_pending()
+        time.sleep(1)