import gradio as gr
from gradio_client import Client

def text_to_image(prompt):
    client = Client("hysts/SDXL")
    img = client.predict(prompt, api_name="/predict")
    return img

@spaces.GPU
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()