restart / app.py
davanstrien's picture
davanstrien HF staff
Update app.py
b8b578e verified
import gradio as gr
from datetime import datetime
from apscheduler.schedulers.background import BackgroundScheduler
from huggingface_hub import restart_space
import os
HF_TOKEN = os.getenv("HF_TOKEN")
time_start = datetime.now()
def restart():
gr.Info("Restarting space...")
restart_space("davanstrien/restart", token=HF_TOKEN)
with gr.Blocks() as demo:
gr.Markdown(f"Time started: {time_start}")
scheduler = BackgroundScheduler()
job = scheduler.add_job(restart, "interval", minutes=2)
scheduler.start()
demo.launch()