Update canvas.py

canvas.py

@@ -35,7 +35,14 @@ async (pose) => {
   poseMakerEl.setPose(pose);
 }
 """
-
+def generate_images(canvas,prompt):        
+    base64_img = canvas
+    image_data = base64.b64decode(base64_img.split(',')[1])
+    input_img = Image.open(BytesIO(image_data)).convert(
+        'RGB').resize((512, 512))
+    input_img = input_img.filter(ImageFilter.GaussianBlur(radius=2))
+    input_img = get_canny_filter(input_img)
+    return input_image
 def placeholder_fn(axis):
     pass
 
@@ -59,6 +66,6 @@ with gr.Blocks() as b:
                             _js=js_pose_template)
     run_button.click(fn=generate_images,
                      inputs=[canvas, prompt],
-                     outputs=[gallery],
+                     outputs=[gr.Image()],
                      _js=get_js_image)
 b.load()