| import gradio as gr | |
| from gradio_client import Client | |
| def text_to_image(prompt): | |
| client = Client("hysts/SDXL", headers={"x-ip-token": x_ip_token}) | |
| img = client.predict(prompt, api_name="/predict") | |
| return img | |
| def generate(prompt): | |
| return text_to_image(prompt) | |
| with gr.Blocks() as demo: | |
| image = gr.Image() | |
| prompt = gr.Textbox(max_lines=1) | |
| prompt.submit(text_to_image, [prompt], [image]) | |
| demo.launch() | |