Convert images to tensors

app.py

@@ -439,10 +439,10 @@ def generate_ai_bg(input_img, prompt):
 
 def blend_details(input_image, relit_image, masked_image):
     with torch.inference_mode():
-        # Load and resize images
-        # input_image = load_image(input_image_path)
-        # relit_image = load_image(relit_image_path)
-        # masked_image = load_image(masked_image_path)
+        # Convert images to tensors
+        input_image = pil2tensor(input_image)
+        relit_image = pil2tensor(relit_image)
+        masked_image = pil2tensor(masked_image)
 
         # Resize input image
         input_image = torch.nn.functional.interpolate(