Spaces:
Sleeping
Sleeping
File size: 3,112 Bytes
fa71535 9f61c1e b84b8a1 9f61c1e fa71535 ca89467 fa71535 3794395 ca89467 3794395 b84b8a1 98256ec b84b8a1 9f61c1e 3794395 fa71535 ca89467 d0f700c ca89467 fa71535 9b843da |
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 54 55 56 57 |
import gradio as gr
def ui(controller):#get_first_result,get_second_result): #controller):
with gr.Blocks() as ui:
with gr.Row():
with gr.Column():
sketch=gr.Image(sources = 'upload', label='Model image', type = 'pil')
with gr.Column():
improved_sketch_view = gr.Image(type="pil", label="Improved Sketch")
#with gr.Row():
with gr.Column():
result=gr.Image(type="pil", label="Colored Image")
with gr.Row():
with gr.Column():
first_prompt = gr.Textbox(label="Prompt", lines=3)
first_negative_prompt = gr.Textbox(label="Negative prompt", lines=3, value="sketch, lowres, error, extra digit, fewer digits, cropped, worst quality,low quality, normal quality, jpeg artifacts, blurry")
firts_prompt_helper = gr.Button(value="Prompt Helper", variant="primary")
#controlnet_scale = gr.Slider(minimum=0.5, maximum=1.25, value=0.5, step=0.01, label="Contr")
improve_sketch = gr.Button(value="Improve Sketch", variant="primary")
with gr.Column():
second_prompt = gr.Textbox(label="Prompt", lines=3)
second_negative_prompt = gr.Textbox(label="Negative prompt", lines=3, value="disfigured, extra digit, fewer digits, cropped, worst quality, low quality")
second_prompt_helper = gr.Button(value="Prompt Helper", variant="primary")
result_button = gr.Button(value="Paint It", variant="primary")
with gr.Row():
gr.Examples(examples=[["./examples/sketch.png",
"solo, open mouth, white background, standing, tail, full body,bmonochrome, greyscale, from side, no humans, animal, dog, animal focus, realistic, walk on the street, add city on the backstage, city street",
"solo, open mouth, simple background, grey background, standing, full body, monochrome, greyscale, from side, no humans, animal, dog, animal focus, lineart, wolf"]],
inputs=[sketch, first_prompt, second_prompt],
label='Examples. Be concrete as much as possible.')
improve_sketch.click(fn=controller.get_first_result,
inputs=[sketch, first_prompt, first_negative_prompt],
outputs=improved_sketch_view)
result_button.click(fn=controller.get_second_result,
inputs=[improved_sketch_view, second_prompt, second_negative_prompt],
outputs=result)
firts_prompt_helper.click(fn=controller.get_help_w_prompt,
inputs=[sketch],
outputs=first_prompt)
second_prompt_helper.click(fn=controller.get_help_w_prompt,
inputs=[improved_sketch_view],
outputs=second_prompt)
ui.queue().launch(debug=True) |