zerotest / app.py
abidlabs's picture
abidlabs HF Staff
Update app.py
41944f2 verified
raw
history blame
406 Bytes
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()