Update app.py

app.py

@@ -39,6 +39,15 @@ async (pose) => {
   poseMakerEl.setPose(pose);
 }
 """
+def get_canny_filter(image):
+    if not isinstance(image, np.ndarray):
+        image = np.array(image) 
+        
+    image = cv2.Canny(image, low_threshold, high_threshold)
+    image = image[:, :, None]
+    image = np.concatenate([image, image, image], axis=2)
+    canny_image = Image.fromarray(image)
+    return canny_image
 def generate_images(canvas):        
     base64_img = canvas
     image_data = base64.b64decode(base64_img.split(',')[1])
@@ -50,8 +59,6 @@ def generate_images(canvas):
 def placeholder_fn(axis):
     pass
 
-
-
 with gr.Blocks() as b:
     with gr.Row():
         canvas = gr.HTML(canvas_html, elem_id="canvas_html", visible=True)
@@ -59,6 +66,7 @@ with gr.Blocks() as b:
             rotation_axis = gr.Radio(choices=["x", "y", "z"], value="x", label="Joint rotation axis")
             pose_template = gr.Radio(choices=["regular", "ballet", "handstand", "split", "kick", "chilling"], value="regular", label="Pose template")
             run_button = gr.Button("Generate")
+            out_im = gr.Image()
     
     rotation_axis.change(fn=placeholder_fn,
                             inputs=[rotation_axis],
@@ -72,7 +80,7 @@ with gr.Blocks() as b:
                             _js=js_pose_template)
     run_button.click(fn=generate_images,
                      inputs=[canvas],
-                     outputs=[gr.Image()],
+                     outputs=[out_im)],
                      _js=get_js_image)
     b.load(None,None,None,_js=load_js)