zerotest / app.py
abidlabs's picture
abidlabs HF Staff
Update app.py
82fdd29 verified
raw
history blame
420 Bytes
import gradio as gr
from gradio_client import Client
import spaces
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()