|
import gradio as gr |
|
import os |
|
import sys |
|
from pathlib import Path |
|
import random |
|
import string |
|
|
|
text_gen=gr.Interface.load("spaces/Omnibus/MagicPrompt-Stable-Diffusion") |
|
|
|
|
|
proc1=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0") |
|
proc2=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0") |
|
proc3=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0") |
|
proc4=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0") |
|
|
|
proc5=gr.Interface.load("models/dreamlike-art/dreamlike-diffusion-1.0") |
|
proc6=gr.Interface.load("models/dreamlike-art/dreamlike-diffusion-1.0") |
|
proc7=gr.Interface.load("models/dreamlike-art/dreamlike-diffusion-1.0") |
|
proc8=gr.Interface.load("models/dreamlike-art/dreamlike-diffusion-1.0") |
|
|
|
def add_random_noise(prompt): |
|
noise = "".join(random.choices(string.ascii_letters + string.punctuation, k=random.randint(1, 10))) |
|
return prompt + noise |
|
|
|
def send_it1(inputs,proc1=proc1): |
|
prompt_with_noise = add_random_noise(inputs) |
|
output1=proc1(prompt_with_noise) |
|
return output1 |
|
|
|
def send_it2(inputs,proc2=proc2): |
|
prompt_with_noise = add_random_noise(inputs) |
|
output2=proc2(prompt_with_noise) |
|
return output2 |
|
|
|
def send_it3(inputs,proc3=proc3): |
|
prompt_with_noise = add_random_noise(inputs) |
|
output3=proc3(prompt_with_noise) |
|
return output3 |
|
|
|
def send_it4(inputs,proc4=proc4): |
|
prompt_with_noise = add_random_noise(inputs) |
|
output4=proc4(prompt_with_noise) |
|
return output4 |
|
|
|
def send_it5(inputs,proc5=proc5): |
|
prompt_with_noise = add_random_noise(inputs) |
|
output5=proc5(prompt_with_noise) |
|
return output5 |
|
|
|
def send_it6(inputs,proc6=proc6): |
|
prompt_with_noise = add_random_noise(inputs) |
|
output6=proc6(prompt_with_noise) |
|
return output6 |
|
|
|
def send_it7(inputs,proc7=proc7): |
|
prompt_with_noise = add_random_noise(inputs) |
|
output7=proc7(prompt_with_noise) |
|
return output7 |
|
|
|
def send_it8(inputs,proc8=proc8): |
|
prompt_with_noise = add_random_noise(inputs) |
|
output8=proc8(prompt_with_noise) |
|
return output8 |
|
|
|
def get_prompts(prompt_text): |
|
return text_gen(prompt_text) |
|
|
|
|
|
with gr.Blocks() as myface: |
|
with gr.Row(): |
|
|
|
input_text=gr.Textbox(label="Short Prompt") |
|
see_prompts=gr.Button("Magic Prompt") |
|
with gr.Row(): |
|
|
|
prompt=gr.Textbox(label="Prettified Prompt") |
|
run=gr.Button("Launch") |
|
|
|
|
|
with gr.Row(): |
|
output1=gr.Image(label="DreamlikeArt PhotoReal 1") |
|
output2=gr.Image(label="DreamlikeArt PhotoReal 2") |
|
with gr.Row(): |
|
output3=gr.Image(label="DreamlikeArt PhotoReal 3") |
|
output4=gr.Image(label="DreamlikeArt PhotoReal 4") |
|
with gr.Row(): |
|
output5=gr.Image(label="DreamlikeArt Diffusion 1") |
|
output6=gr.Image(label="DreamlikeArt Diffusion 2") |
|
with gr.Row(): |
|
output7=gr.Image(label="DreamlikeArt Diffusion 3") |
|
output8=gr.Image(label="DreamlikeArt Diffusion 4") |
|
|
|
run.click(send_it1, inputs=[prompt], outputs=[output1]) |
|
run.click(send_it2, inputs=[prompt], outputs=[output2]) |
|
run.click(send_it3, inputs=[prompt], outputs=[output3]) |
|
run.click(send_it4, inputs=[prompt], outputs=[output4]) |
|
run.click(send_it5, inputs=[prompt], outputs=[output5]) |
|
run.click(send_it6, inputs=[prompt], outputs=[output6]) |
|
run.click(send_it7, inputs=[prompt], outputs=[output7]) |
|
run.click(send_it8, inputs=[prompt], outputs=[output8]) |
|
see_prompts.click(get_prompts, inputs=[input_text], outputs=[prompt]) |
|
|
|
|
|
myface.queue(concurrency_count=2) |
|
myface.launch(enable_queue=True, inline=True) |