Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -67,7 +67,7 @@ def generate_images(canvas,output_type):
|
|
67 |
input_img = input_img.filter(ImageFilter.GaussianBlur(radius=2))
|
68 |
input_img = ImageOps.invert(get_canny_filter(input_img)) if output_type == "Canny" else input_img
|
69 |
input_img = ImageOps.invert(input_img) if output_type == "Depth" else input_img
|
70 |
-
input_img = Image.open(BytesIO(base64_img)).resize((512,512)).crop((left, top, right, bottom)) if output_type == "Base" else input_img
|
71 |
|
72 |
#input_img = ImageOps.invert(input_img)
|
73 |
input_img.save(f"{uid}-pose.png")
|
@@ -82,7 +82,7 @@ with gr.Blocks() as b:
|
|
82 |
with gr.Column():
|
83 |
rotation_axis = gr.Radio(choices=["x", "y", "z"], value="x", label="Joint rotation axis")
|
84 |
pose_template = gr.Radio(choices=["regular", "ballet", "handstand", "split", "kick", "chilling"], value="regular", label="Pose template")
|
85 |
-
output_type = gr.Radio(choices=["
|
86 |
run_button = gr.Button("Generate")
|
87 |
out_text = gr.Textbox(label="Image URL")
|
88 |
out_file=gr.File()
|
|
|
67 |
input_img = input_img.filter(ImageFilter.GaussianBlur(radius=2))
|
68 |
input_img = ImageOps.invert(get_canny_filter(input_img)) if output_type == "Canny" else input_img
|
69 |
input_img = ImageOps.invert(input_img) if output_type == "Depth" else input_img
|
70 |
+
#input_img = Image.open(BytesIO(base64_img)).resize((512,512)).crop((left, top, right, bottom)) if output_type == "Base" else input_img
|
71 |
|
72 |
#input_img = ImageOps.invert(input_img)
|
73 |
input_img.save(f"{uid}-pose.png")
|
|
|
82 |
with gr.Column():
|
83 |
rotation_axis = gr.Radio(choices=["x", "y", "z"], value="x", label="Joint rotation axis")
|
84 |
pose_template = gr.Radio(choices=["regular", "ballet", "handstand", "split", "kick", "chilling"], value="regular", label="Pose template")
|
85 |
+
output_type = gr.Radio(choices=["Depth", "Canny"], value="Depth", label="Image Output")
|
86 |
run_button = gr.Button("Generate")
|
87 |
out_text = gr.Textbox(label="Image URL")
|
88 |
out_file=gr.File()
|