Add debug messages, switch to non-gpu rembg

marble.py

@@ -227,12 +227,13 @@ def run_parametric_control(
     texture_image: Image.Image = None,
     num_inference_steps: int = 30,
     seed: int = 42,
-    depth_map: Image.Image = None,
-    mask: Image.Image = None,
+    depth_map: Optional[Image.Image] = None,
+    mask: Optional[Image.Image] = None,
 ) -> Image.Image:
     """Run parametric control with metallic and roughness adjustments."""
     # Get depth map
     if depth_map is None:
+        print("No depth map provided, running DPT depth estimation")
         model, processor = get_dpt_model()
         depth_map = run_dpt_depth(target_image, model, processor)
     else:
@@ -240,16 +241,19 @@ def run_parametric_control(
 
     # Prepare mask and init image
     if mask is None:
+        print("No mask provided, preparing mask")
         mask = prepare_mask(target_image)
     else:
         mask = mask.resize((1024, 1024))
 
+    print("Preparing initial image")
     if texture_image is None:
         texture_image = target_image
 
     init_img = prepare_init_image(target_image, mask)
 
     # Generate edit
+    print("Generating parametric edit")
     images = ip_model.generate_parametric_edits(
         texture_image,
         image=init_img,
@@ -274,12 +278,13 @@ def run_blend(
     edit_strength: float = 0.0,
     num_inference_steps: int = 20,
     seed: int = 1,
-    depth_map: Image.Image = None,
-    mask: Image.Image = None,
+    depth_map: Optional[Image.Image] = None,
+    mask: Optional[Image.Image] = None,
 ) -> Image.Image:
     """Run blending between two texture images."""
     # Get depth map
     if depth_map is None:
+        print("No depth map provided, running DPT depth estimation")
         model, processor = get_dpt_model()
         depth_map = run_dpt_depth(target_image, model, processor)
     else:
@@ -287,12 +292,16 @@ def run_blend(
 
     # Prepare mask and init image
     if mask is None:
+        print("No mask provided, preparing mask")
         mask = prepare_mask(target_image)
     else:
         mask = mask.resize((1024, 1024))
+
+    print("Preparing initial image")
     init_img = prepare_init_image(target_image, mask)
 
     # Generate edit
+    print("Generating edit")
     images = ip_model.generate_edit(
         start_image=texture_image1,
         pil_image=texture_image1,

requirements.txt

@@ -1,5 +1,5 @@
 diffusers==0.30
-rembg[gpu]
+rembg
 einops==0.7.0
 transformers==4.27.4
 opencv-python==4.7.0.68