File size: 1,367 Bytes
5aebe57
 
 
24277b9
5aebe57
 
 
 
24277b9
5aebe57
 
 
24277b9
 
 
5aebe57
 
 
 
d315355
24277b9
5aebe57
 
24277b9
5aebe57
 
 
 
 
 
 
88e3ee7
 
5aebe57
24277b9
334740e
24277b9
 
5aebe57
 
24277b9
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
import gradio as gr
from utils import *

with gr.Blocks() as gr_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="Illustration Style"
        )

        method_dropdown = gr.Dropdown(
            label="Select Guidance Type",
            choices=["edge", "contrast", "sharpness", "blue", "brightness"],
            value="contrast"
        )

        inputs = [text_input, concept_dropdown, method_dropdown]

    with gr.Row():
        outputs = gr.Gallery(
            label="Generative Images", show_label=True,
            columns=[2], rows=[1], object_fit="contain"
        )

    # with gr.Row():
    #     gr.Examples(examples=get_examples(), inputs=inputs, outputs=outputs, fn=show_image, cache_examples=True)

title = "Generative Art - Stable Diffusion with Styles and Custom Loss"
demo = gr.Interface(fn=show_image, inputs=inputs, outputs=outputs, title=title)



if __name__ == "__main__":
    demo.launch(enable_queue=True, debug=True)