import gradio as gr
# from gradio_client import Client

# def text_to_image(client, prompt):
#     img = client.predict(prompt, "", "", "", api_name="/run")
#     return img


# def set_client_for_session(request: gr.Request):
#     x_ip_token = request.headers['x-ip-token']

#     # The "gradio/text-to-image" space is a ZeroGPU space
#     return Client("hysts/SDXL", headers={"x-ip-token": x_ip_token})

# with gr.Blocks() as demo:
#     client = gr.State()
#     image = gr.Image()
#     prompt = gr.Textbox(max_lines=1)

#     prompt.submit(text_to_image, [client, prompt], [image])

#     demo.load(set_client_for_session, None, client)
# demo.launch()


demo = gr.load("hysts/SDXL")
demo.launch()