Update app.py

app.py

@@ -60,11 +60,15 @@ def generate_images(canvas,output_type):
     top = 0
     right = 384
     bottom = 512   
+
     input_img = Image.open(BytesIO(image_data)).convert(
         'RGB').resize((512,512)).crop((left, top, right, bottom))
 
     input_img = input_img.filter(ImageFilter.GaussianBlur(radius=2))
-    input_img = get_canny_filter(input_img) if output_type == "Canny" else input_img
+    input_img = ImageOps.invert(get_canny_filter(input_img)) if output_type == "Canny" else input_img
+    input_img = ImageOps.invert(input_img) if output_type == "Depth" else input_img
+    input_img = canvas.resize((512,512)).crop((left, top, right, bottom)) if output_type == "Base" else input_img
+    
     #input_img = ImageOps.invert(input_img)
     input_img.save(f"{uid}-pose.png")
     out = os.path.abspath(f"{uid}-pose.png")