Gainward777 commited on
Commit
dbebde5
·
verified ·
1 Parent(s): 7d1980b

Update ui/gradio_ui.py

Browse files
Files changed (1) hide show
  1. ui/gradio_ui.py +5 -5
ui/gradio_ui.py CHANGED
@@ -3,22 +3,22 @@ from PIL import Image
3
  import numpy as np
4
 
5
 
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
14
- SER_SKETCH=img
15
 
16
  def ui(controller):#get_first_result,get_second_result): #controller):
17
  with gr.Blocks() as ui:
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(type="pil")#visible=False)
22
 
23
  with gr.Column():
24
  improved_sketch_view = gr.Image(type="pil", label="Improved Sketch")
 
3
  import numpy as np
4
 
5
 
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
14
+ #SER_SKETCH=img
15
 
16
  def ui(controller):#get_first_result,get_second_result): #controller):
17
  with gr.Blocks() as ui:
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(type="pil", visible=False)
22
 
23
  with gr.Column():
24
  improved_sketch_view = gr.Image(type="pil", label="Improved Sketch")