|
import gradio as gr |
|
import requests |
|
import os |
|
import random |
|
|
|
|
|
|
|
|
|
name2 = "runwayml/stable-diffusion-v1-5" |
|
models=[ |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
] |
|
o = os.getenv("P") |
|
|
|
def ac(): |
|
def im_fn(put,fac="",h=None): |
|
if h == o: |
|
rn = random.randint(0, 19) |
|
model=models[rn] |
|
put = f"{put}{fac}" |
|
fac = f"{fac} " |
|
print (rn) |
|
return model(put),fac |
|
elif h != o: |
|
return(None,None) |
|
def cl_fac(): |
|
return "" |
|
with gr.Blocks(css='css.css') 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() |
|
with gr.Row(visible=False): |
|
h=gr.Textbox("Q") |
|
fac_b = gr.Textbox(value="",visible=False) |
|
btn1.click(cl_fac,None,fac_b) |
|
btn1.click(im_fn,[put,fac_b,h],[out1,fac_b]) |
|
out1.change(im_fn,[put,fac_b,h],[out2,fac_b]) |
|
out2.change(im_fn,[put,fac_b,h],[out3,fac_b]) |
|
out3.change(im_fn,[put,fac_b,h],[out4,fac_b]) |
|
b.queue(concurrency_count=100).launch(show_api=False) |
|
ac() |