File size: 597 Bytes
2dbca95 06c9b54 2dbca95 b6cc84b 709bade d10c852 6275c42 a1b49d8 06c9b54 6275c42 06c9b54 4645e6a 6275c42 06c9b54 6275c42 06c9b54 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
import gradio as gr
from gradio_client import Client
def set_client_for_session(request: gr.Request):
x_ip_token = request.headers['x-ip-token']
return Client("hysts/SDXL", headers={"x-ip-token": x_ip_token})
def text_to_image(prompt, request: gr.Request):
client = set_client_for_session(request)
img = client.predict(prompt, "", "", "", api_name="/run")
return img
with gr.Blocks() as demo:
image = gr.Image()
prompt = gr.Textbox(max_lines=1)
prompt.submit(text_to_image, [prompt], [image])
demo.launch()
# demo = gr.load("spaces/hysts/SDXL")
# demo.launch() |