app.py

import gradio as gr
from gradio_client import Client

def text_to_image(client, prompt):
    img = client.predict(prompt, api_name="/predict")
    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("gradio/text-to-image", 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()