Update app.py

app.py

@@ -8,6 +8,8 @@ from huggingface_hub import HfApi, Repository
 
 from onnx_export import convert
 
+from apscheduler.schedulers.background import BackgroundScheduler
+
 DATASET_REPO_URL = "https://huggingface.co/datasets/optimum/exporters"
 DATA_FILENAME = "data.csv"
 DATA_FILE = os.path.join("data", DATA_FILENAME)
@@ -151,4 +153,11 @@ with gr.Blocks() as demo:
         outputs=output,
     )
 
+def restart_space():
+    HfApi().restart_space(repo_id="onnx/export", token=HF_TOKEN, factory_reboot=True)
+
+scheduler = BackgroundScheduler()
+scheduler.add_job(restart_space, "interval", seconds=21600)
+scheduler.start()
+
 demo.launch()

requirements.txt

@@ -2,3 +2,4 @@ torch==2.1.2
 transformers==4.37.2
 huggingface_hub==0.20.3
 optimum[onnxruntime,exporters]==1.16.2
+APScheduler