import gradio as gr
import os
from huggingface_hub import HfApi

hf_token = os.getenv('hf_token')
repo_id = "duypro247/TestScrape"
api = HfApi()

def restart_space():
    space_runtime = api.restart_space(repo_id=repo_id, token=hf_token)

demo = gr.Blocks()

with demo:
    text = gr.Textbox()
    b1 = gr.Button("Restart Space")
    b1.click(restart_space)

demo.launch()