Update app.py

app.py

@@ -199,10 +199,13 @@ def get_image(img_path: str, mask: bool=False, scale: bool=True) -> np.array:
         _, binary_mask = cv2.threshold(img, 127, 255, cv2.THRESH_BINARY)
         return np.where(binary_mask == 255, 1, 0)
     
+    if img.shape[-1] == 3:  # Check if the image has 3 channels
+        img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+    
     if scale:
-        return cv2.cvtColor(img, cv2.COLOR_BGR2RGB).astype('double') / 255.0
+        return img.astype('double') / 255.0
     
-    return cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+    return img
 
 def blend_images(bg_img, obj_img, mask_img, method):
     bg_img = get_image(bg_img)