Spaces:
Sleeping
Sleeping
File size: 630 Bytes
2dbca95 6275c42 2dbca95 6275c42 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
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()
|