text-to-image / app.py
xp3857's picture
Update app.py
184a373
raw
history blame
1.06 kB
import gradio as gr
import requests
import os
name2 = "runwayml/stable-diffusion-v1-5"
model = gr.Interface.load(f"models/{name2}")
o = os.getenv("P")
h = "Q"
def ac():
def im_fn(put,fac=""):
if h == o:
put = f"{put}{fac}"
fac = f"{fac} "
return model(put),fac
elif h != o:
return(None,None)
def cl_fac():
return ""
with gr.Blocks() as b:
with gr.Row():
put = gr.Textbox()
btn1 = gr.Button()
with gr.Row():
out1 = gr.Image()
out2 = gr.Image()
with gr.Row():
out3 = gr.Image()
out4 = gr.Image()
fac_b = gr.Textbox(value="",visible=False)
btn1.click(cl_fac,None,fac_b)
btn1.click(im_fn,[put,fac],[out1,fac_b])
out1.change(im_fn,[put,fac],[out2,fac_b])
out2.change(im_fn,[put,fac],[out3,fac_b])
out3.change(im_fn,[put,fac],[out4,fac_b])
b.queue(concurrency_count=100).launch()
ac()