no scheduler

app.py

@@ -69,17 +69,22 @@ def predict(dict, depth, prompt="", negative_prompt="", guidance_scale=7.5, step
         negative_prompt = None
     scheduler_class_name = scheduler.split("-")[0]
 
-    init_image = cv2.resize(dict["image"], (512, 512))
+    init_image = cv2.cvtColor(cv2.resize(dict["image"], (512, 512)), cv2.COLOR_BGR2RGB)
+    cv2.imwrite("temp.jpg", init_image)
+
     mask = Image.fromarray(cv2.resize(dict["mask"], (512, 512))[:,:,0])
+    mask.save("temp_mask.jpg")
 
     if (depth is None):
         depth_image = estimate_depth(init_image)
+        
     else:
         depth_image = depth
         depth_image = Image.fromarray(depth_image[:,:,0].astype("uint8"))
 
-    scheduler = getattr(diffusers, scheduler_class_name)
-    pipe.scheduler = scheduler.from_pretrained("Intel/ldm3d-4c", subfolder="scheduler")
+    depth_image.save("temp_depth.jpg")
+    #scheduler = getattr(diffusers, scheduler_class_name)
+    #pipe.scheduler = scheduler.from_pretrained("Intel/ldm3d-4c", subfolder="scheduler")
     
     
     depth_image = depth_image.resize((512, 512))