app

app.py

@@ -70,7 +70,6 @@ def predict(dict, depth, prompt="", negative_prompt="", guidance_scale=7.5, step
     scheduler_class_name = scheduler.split("-")[0]
 
     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")
@@ -82,6 +81,9 @@ def predict(dict, depth, prompt="", negative_prompt="", guidance_scale=7.5, step
         depth_image = depth
         depth_image = Image.fromarray(depth_image[:,:,0].astype("uint8"))
 
+    init_image = Image.fromarray(init_image.astype("uint8"))
+    init_image.save("temp_image.jpg")
+    
     depth_image.save("temp_depth.jpg")
     #scheduler = getattr(diffusers, scheduler_class_name)
     #pipe.scheduler = scheduler.from_pretrained("Intel/ldm3d-4c", subfolder="scheduler")