File size: 3,914 Bytes
b6b08f6 af2f25b b593cfa 477a5c9 7416ccf b593cfa f97094c b593cfa f97094c 477a5c9 b593cfa bd88d47 82dac02 c30ef6f 7b3209b db124c0 81825d2 82dac02 bd88d47 a2ffcba 510f059 82dac02 4b9f915 b593cfa 4b9f915 f16e6a1 4b9f915 82dac02 4b9f915 bd88d47 b593cfa 79fd217 35026e1 184a373 82dac02 b593cfa 1064155 184a373 dfd7221 3c896ec 0ec4659 3c896ec d3fb471 b6eb1a4 184a373 b6eb1a4 184a373 4e98d1d f0174d9 b593cfa 4b9f915 0ec4659 f35081a 184a373 176dcc8 962ece5 79fd217 82dac02 79fd217 962ece5 b39b122 176dcc8 8646ee7 af2f25b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 |
import gradio as gr
import os
import requests
import random
import time
r = requests.get(f'https://huggingface.co/spaces/xp3857/text-to-image/raw/main/css.css')
css = r.text
name2 = "runwayml/stable-diffusion-v1-5"
model = gr.Interface.load(f"models/{name2}")
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")
o = "V"
m_out = ("""
<div id="restart">
<h3 id="head">Loading Time Limit Reached.</h3><br>
<h4 id="cont">Please choose a Simpler Prompt, or <b>Upgrade</b> for faster loading.</h4>
</div>
""")
def ac():
def clear():
return gr.update(value=0),gr.update(value=0)
def start():
stamp = time.time()
return gr.update(value=stamp),gr.update(value=0)
def end(stamp):
ts = stamp + 120
ti = time.time()
if ti > ts and stamp != 0:
return gr.update(value=1),gr.HTML.update(f"{m_out}", visible=True)
else:
return gr.update(value=0),None
def im_fn(put,fac="",h=None):
try:
if h == o:
put = f"{put}{fac}"
fac = f"{fac} "
rn = random.randint(0, 19)
model=models[rn]
return model(put),fac
elif h != o:
return(None,None)
except Exception:
return None, None
def cl_fac():
return "",gr.HTML.update("", visible=False)
with gr.Blocks(css=css) as b:
gr.Progress(im_fn)
with gr.Row():
with gr.Column():
put = gr.Textbox()
with gr.Column():
with gr.Row():
btn1 = gr.Button("Run")
btn2 = gr.Button("Clear",style="background:blue;")
message=gr.HTML("", visible=False)
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(value="V")
t_state=gr.Number()
t_switch=gr.Textbox(value=0)
def clear_all():
return "",None,None,None,None,None,None,1
fac_b = gr.Textbox(value="",visible=False)
def noth():
return
a1=btn1.click(noth,None,btn1,every=1)
btn1.click(cl_fac,None,[fac_b,message])
b1=btn1.click(start,None,[t_state,t_switch])
sta = t_state.change(end,t_state,[t_switch,message],every=1,show_progress=False)
b2=btn1.click(im_fn,[put,fac_b,h],[out1,fac_b], show_progress=True)
b3=out1.change(im_fn,[put,fac_b,h],[out2,fac_b], show_progress=False)
b4=out2.change(im_fn,[put,fac_b,h],[out3,fac_b], show_progress=False)
b5=out3.change(im_fn,[put,fac_b,h],[out4,fac_b], show_progress=False)
swi=t_switch.change(clear,None,[t_switch,fac_b], cancels=[sta,b2,b3,b4,b5],show_progress=False)
btn2.click(clear_all, None,[fac_b,put,out1,out2,out3,out4,t_state,t_switch],cancels=[b1,sta,b2,b3,b4,b5,swi,a1],show_progress=False)
b.queue(concurrency_count=100).launch(show_api=False)
ac() |