|
|
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() |
|
|
|
|
|
|
|
|
|
