Omnibus commited on
Commit
7286c79
·
1 Parent(s): 3400de0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=["Base", "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()
 
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()