Gainward777 commited on
Commit
5cd8336
·
verified ·
1 Parent(s): 8f6e7d9

Update ui/gradio_ui.py

Browse files
Files changed (1) hide show
  1. 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
- SER_SKETCH=Image.fromarray(img["composite"])
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=[SER_SKETCH, first_prompt, first_negative_prompt], #[sketch, first_prompt, first_negative_prompt],
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=[SER_SKETCH], #[sketch],
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,