Merge branch apscheduler pr from 'main' into pr/30

app.py

@@ -16,6 +16,8 @@ from gradio_logsview.logsview import Log, LogsView, LogsViewRunner
 from mergekit.config import MergeConfiguration
 
 from clean_community_org import garbage_collect_empty_models
+from apscheduler.schedulers.background import BackgroundScheduler
+from datetime import datetime, timezone
 
 has_gpu = torch.cuda.is_available()
 
@@ -255,10 +257,29 @@ def extract(finetuned_model: str, base_model: str, rank: int, hf_token: str, rep
         )
         yield runner.log(f"Lora successfully uploaded to HF: {repo_url.repo_id}")
 
+# This is workaround. As the space always getting stuck.
+def _restart_space():
+    huggingface_hub.HfApi().restart_space(repo_id="arcee-ai/mergekit-gui", token=COMMUNITY_HF_TOKEN, factory_reboot=False)
+# Run garbage collection every hour to keep the community org clean.
+# Empty models might exists if the merge fails abruptly (e.g. if user leaves the Space).
+def _garbage_remover():
+    try:
+        garbage_collect_empty_models(token=COMMUNITY_HF_TOKEN)
+    except Exception as e:
+        print("Error running garbage collection", e)
+
+scheduler = BackgroundScheduler()
+restart_space_job = scheduler.add_job(_restart_space, "interval", seconds=21600)
+garbage_remover_job = scheduler.add_job(_garbage_remover, "interval", seconds=3600)
+scheduler.start()
+next_run_time_utc = restart_space_job.next_run_time.astimezone(timezone.utc)
+
+NEXT_RESTART = f"Next Restart: {next_run_time_utc.strftime('%Y-%m-%d %H:%M:%S')} (UTC)"
 
 with gr.Blocks() as demo:
     gr.Markdown(MARKDOWN_DESCRIPTION)
-
+    gr.Markdown(NEXT_RESTART)
+    
     with gr.Tabs():
         with gr.TabItem("Merge Model"):
             with gr.Row():
@@ -335,7 +356,7 @@ with gr.Blocks() as demo:
 
 
 # Run garbage collection every hour to keep the community org clean.
-# Empty models might exist if the merge fails abruptly (e.g. if user leaves the Space).
+# Empty models might exists if the merge fails abruptly (e.g. if user leaves the Space).
 def _garbage_collect_every_hour():
     while True:
         try:

requirements.txt

@@ -1,3 +1,4 @@
+apscheduler
 torch
 bitsandbytes
 git+https://github.com/arcee-ai/mergekit.git