raghunc0's picture
Update app.py
222f6f5
import gradio as gr
from sd_utils import *
with gr.Blocks() as interface:
#gr.HTML(value=HTML_TEMPLATE, show_label=False)
with gr.Row():
text_input = gr.Textbox(
label="Enter your prompt here",
placeholder="A portrait James bond in the style of ...",
)
concept_dropdown = gr.Dropdown(
label="Select an SD Concept",
choices=["midjourney", "birb", "indian-watercolor", "Marc Allante", "herge"],
value='Dream'
)
seed = gr.Slider(
label="Random Seed",
minimum=0,
maximum=10000,
step=1,
value=42
)
inputs = [text_input, concept_dropdown, seed]
with gr.Row():
outputs = gr.Gallery(
label="Generated Image", show_label=True,
columns=[1], rows=[1], object_fit="contain"
)
with gr.Row():
button = gr.Button("Generate")
button.click(generate_with_embs_custom_loss, inputs=inputs, outputs=outputs)
with gr.Row():
gr.Examples(examples=get_examples(), inputs=inputs, outputs=outputs, fn=generate_art, cache_examples=True)
if __name__ == "__main__":
interface.launch(enable_queue=True)