Spaces:
Runtime error
Runtime error
File size: 3,324 Bytes
063b292 c194fdd 30cee80 f5e64be 932a21a 063b292 f5e64be 0d4e1e9 30cee80 e90a37e 0bb5a0b 063b292 f8a1403 063b292 f8a1403 c194fdd 063b292 c194fdd 063b292 c194fdd 063b292 c194fdd 063b292 c194fdd 063b292 c194fdd 063b292 c194fdd 063b292 c194fdd f8a1403 c194fdd 063b292 c194fdd f8a1403 063b292 f8a1403 c194fdd 063b292 c194fdd 063b292 c194fdd 063b292 c194fdd 063b292 c194fdd 063b292 c194fdd 063b292 c194fdd 063b292 c194fdd |
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 109 110 111 112 113 114 115 116 |
import gradio as gr
import time
models = [
"mann-e/Mann-E_Dreams",
"Yntec/ChickFlick",
"John6666/ultimate-realistic-mix-v2-sdxl",
"Yntec/CrystalReality",
"John6666/9527-detail-realistic-xl-v55mix-sdxl",
"John6666/epicrealism-xl-v8kiss-sdxl",
"John6666/wai-real-mix-v8-sdxl",
"John6666/real-vis-xl-v40-sdxl",
"Yntec/DegreesOfFreedom",
]
model_functions = {}
model_idx = 1
for model_path in models:
try:
model_functions[model_idx] = gr.Interface.load(f"models/{model_path}")
except Exception as error:
def the_fn(txt):
return None
model_functions[model_idx] = gr.Interface(fn=the_fn, inputs=gr.Textbox(), outputs=gr.Image())
model_idx += 1
def send_it_idx(idx):
def send_it_fn(prompt):
output = (model_functions.get(idx) or model_functions.get(1)).predict(prompt)
return output
return send_it_fn
def get_prompts(prompt_text):
return prompt_text
def clear_it(val):
return 0
def all_task_end(cnt, t_stamp):
to = t_stamp + 360
et = time.time()
if et > to and t_stamp != 0:
d = gr.update(value=0)
tog = gr.update(value=1)
else:
d = gr.update(value=et) if cnt != 0 else gr.update(value=0)
tog = gr.update(value=0)
return d, tog
def all_task_start():
t_stamp = time.time()
return gr.update(value=t_stamp), gr.update(value=t_stamp), gr.update(value=0)
def clear_fn():
nn = len(models)
return (None, *[None for _ in range(nn)])
with gr.Interface(title="SD Models") as my_interface:
with gr.Column():
with gr.Row():
with gr.Column():
primary_prompt = gr.Textbox(label="Prompt", default="")
with gr.Column():
run = gr.Button("Run")
clear_btn = gr.Button("Clear")
with gr.Row():
sd_outputs = {}
model_idx = 1
for model_path in models:
with gr.Column():
sd_outputs[model_idx] = gr.Image(label=model_path)
model_idx += 1
start_box = gr.Number(visible=False)
end_box = gr.Number(visible=False)
tog_box = gr.Textbox(default=0, visible=False)
def on_change_start_box(value):
return all_task_end(value, end_box.value)
start_box.change(
on_change_start_box,
inputs=[start_box],
outputs=[start_box, tog_box],
every=1
)
primary_prompt.submit(all_task_start, None, [start_box, end_box, tog_box])
run.click(all_task_start, None, [start_box, end_box, tog_box])
runs_dict = {}
model_idx = 1
for model_path in models:
runs_dict[model_idx] = run.click(
send_it_idx(model_idx),
inputs=[primary_prompt],
outputs=[sd_outputs[model_idx]]
)
model_idx += 1
clear_btn.click(
clear_fn,
None,
[primary_prompt, *list(sd_outputs.values())],
cancels=[*list(runs_dict.values())]
)
tog_box.change(
clear_it,
tog_box,
tog_box,
cancels=[*list(runs_dict.values())]
)
my_interface.queue(concurrency_count=1)
my_interface.launch()
|