Spaces:
Sleeping
Sleeping
Update ui/gradio_ui.py
Browse files- 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"
|
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")
|