Spaces:
Running
Running
File size: 1,774 Bytes
5aebe57 d315355 5aebe57 |
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 |
import gradio as gr
from utils import *
styles_mapping = {
"Illustration Style": '<illustration-style>', "Line Art":'<line-art>',
"Hitokomoru Style":'<hitokomoru-style-nao>', "Marc Allante": '<Marc_Allante>',
"Midjourney":'<midjourney-style>', "Hanfu Anime": '<hanfu-anime-style>',
"Birb Style": '<birb-style>'
}
with gr.Blocks() as interface:
#gr.HTML(value=HTML_TEMPLATE, show_label=False)
with gr.Row():
text_input = gr.Textbox(
label="Enter your prompt",
placeholder="Cats fighting on the road.....",
)
concept_dropdown = gr.Dropdown(
label="Select a Concept",
choices=["Illustration Style", "Line Art", "Hitokomoru Style", "Marc Allante", "Midjourney", "Hanfu Anime", "Birb Style"],
value='Marc Allante'
)
method_dropdown = gr.Dropdown(
label="Select Guidance Type",
choices=["edge", "contrast", "sharpness", "blue", "brightness"],
value='Contrast'
)
seed_slider = gr.Slider(
label="Random Seed",
minimum=0,
maximum=2000,
step=1,
value=42
)
inputs = [text_input, concept_dropdown, method_dropdown, seed_slider]
with gr.Row():
outputs = gr.Gallery(
label="Generative Images", show_label=True,
columns=[2], rows=[1], object_fit="contain"
)
with gr.Row():
button = gr.Button("Generate Image")
button.click(show_image, inputs=inputs, outputs=outputs)
with gr.Row():
gr.Examples(examples=get_examples(), inputs=inputs, outputs=outputs, fn=show_image, cache_examples=True)
if __name__ == "__main__":
interface.launch(enable_queue=True) |