Spaces:
Sleeping
Sleeping
Update ui/gradio_ui.py
Browse files- ui/gradio_ui.py +6 -5
ui/gradio_ui.py
CHANGED
@@ -6,8 +6,8 @@ import numpy as np
|
|
6 |
SER_SKETCH=Image.fromarray(np.zeros((1024,1024)))
|
7 |
|
8 |
def get_img_from_sketchpad(img):
|
9 |
-
global SER_SKETCH
|
10 |
-
|
11 |
|
12 |
def set_ser_sketch(img):
|
13 |
global SER_SKETCH
|
@@ -18,6 +18,7 @@ def ui(controller):#get_first_result,get_second_result): #controller):
|
|
18 |
with gr.Row():
|
19 |
with gr.Column():
|
20 |
sketch=gr.Sketchpad(sources = 'upload', label='Sketch', type='numpy', image_mode='L', brush=gr.Brush()) #gr.Image(sources = 'upload', label='Sketch', type = 'pil')
|
|
|
21 |
|
22 |
with gr.Column():
|
23 |
improved_sketch_view = gr.Image(type="pil", label="Improved Sketch")
|
@@ -48,10 +49,10 @@ def ui(controller):#get_first_result,get_second_result): #controller):
|
|
48 |
inputs=[sketch, first_prompt, second_prompt],
|
49 |
label='Examples. Be concrete as much as possible.')
|
50 |
|
51 |
-
sketch.input(fn=get_img_from_sketchpad, inputs=sketch, show_progress="hidden")
|
52 |
|
53 |
improve_sketch.click(fn=controller.get_first_result,
|
54 |
-
inputs=[
|
55 |
outputs=improved_sketch_view)
|
56 |
|
57 |
result_button.click(fn=controller.get_second_result,
|
@@ -59,7 +60,7 @@ def ui(controller):#get_first_result,get_second_result): #controller):
|
|
59 |
outputs=result)
|
60 |
|
61 |
firts_prompt_helper.click(fn=controller.get_help_w_prompt,
|
62 |
-
inputs=[
|
63 |
outputs=first_prompt)
|
64 |
|
65 |
second_prompt_helper.click(fn=controller.get_help_w_prompt,
|
|
|
6 |
SER_SKETCH=Image.fromarray(np.zeros((1024,1024)))
|
7 |
|
8 |
def get_img_from_sketchpad(img):
|
9 |
+
#global SER_SKETCH
|
10 |
+
return Image.fromarray(img["composite"])
|
11 |
|
12 |
def set_ser_sketch(img):
|
13 |
global SER_SKETCH
|
|
|
18 |
with gr.Row():
|
19 |
with gr.Column():
|
20 |
sketch=gr.Sketchpad(sources = 'upload', label='Sketch', type='numpy', image_mode='L', brush=gr.Brush()) #gr.Image(sources = 'upload', label='Sketch', type = 'pil')
|
21 |
+
ser_image=gr.Image()#visible=False)
|
22 |
|
23 |
with gr.Column():
|
24 |
improved_sketch_view = gr.Image(type="pil", label="Improved Sketch")
|
|
|
49 |
inputs=[sketch, first_prompt, second_prompt],
|
50 |
label='Examples. Be concrete as much as possible.')
|
51 |
|
52 |
+
sketch.input(fn=get_img_from_sketchpad, inputs=sketch, output=ser_image, show_progress="hidden")
|
53 |
|
54 |
improve_sketch.click(fn=controller.get_first_result,
|
55 |
+
inputs=[ser_image, first_prompt, first_negative_prompt], #[sketch, first_prompt, first_negative_prompt],
|
56 |
outputs=improved_sketch_view)
|
57 |
|
58 |
result_button.click(fn=controller.get_second_result,
|
|
|
60 |
outputs=result)
|
61 |
|
62 |
firts_prompt_helper.click(fn=controller.get_help_w_prompt,
|
63 |
+
inputs=[ser_image], #[sketch],
|
64 |
outputs=first_prompt)
|
65 |
|
66 |
second_prompt_helper.click(fn=controller.get_help_w_prompt,
|