This commit is contained in:
lvmin 2023-08-10 13:59:40 -07:00
parent 14390356dd
commit f20cc1967e

27
draft.py Normal file
View File

@ -0,0 +1,27 @@
import gradio as gr
from modules.sdxl_styles import apply_style
from modules.default_pipeline import process
def generate_clicked(positive_prompt):
p, n = apply_style('sai-cinematic', positive_prompt, '')
print(p)
print(n)
return process(positive_prompt=p,
negative_prompt=n)
block = gr.Blocks().queue()
with block:
with gr.Column():
prompt = gr.Textbox(label="Prompt", value='a handsome man in forest')
run_button = gr.Button(label="Run")
result_gallery = gr.Gallery(label='Output', show_label=False, elem_id="gallery", height=1280)
run_button.click(fn=generate_clicked, inputs=[prompt], outputs=[result_gallery])
block.launch()