bp

app.py

@@ -156,6 +156,7 @@ def depth_moge(input_img):
     output = moge_model.infer(input_img_torch).cpu()
     return output 
 
+@GPU
 def mask_rembg(input_img):
     #masked_img = rembg.remove(input_img,)
     output_img = rembg.remove(input_img, alpha_matting=False, post_process_mask=True)