Update vtoonify_model.py

vtoonify_model.py

@@ -142,16 +142,15 @@ class Model():
         return frame, instyle, message
     
     #@torch.inference_mode()
-    def detect_and_align_image(self, image: str, top: int, bottom: int, left: int, right: int
-                              ) -> tuple[np.ndarray, torch.Tensor, str]:
-        if image is None:
-            return np.zeros((256,256,3), np.uint8), None, 'Error: fail to load empty file.'
-        frame = cv2.imread(image)
-        if frame is None:
-            return np.zeros((256,256,3), np.uint8), None, 'Error: fail to load the image.'       
-        frame = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR)
-        return self.detect_and_align(frame, top, bottom, left, right)
-    
+    def detect_and_align_image(self, frame_rgb: np.ndarray, top: int, bottom: int, left: int, right: int) -> tuple[np.ndarray, torch.Tensor, str]:
+    if frame_rgb is None:
+        return np.zeros((256, 256, 3), np.uint8), None, 'Error: fail to load the image.'
+
+    # Convert RGB to BGR
+    frame_bgr = cv2.cvtColor(frame_rgb, cv2.COLOR_RGB2BGR)
+
+    return self.detect_and_align(frame_bgr, top, bottom, left, right)
+
     def detect_and_align_video(self, video: str, top: int, bottom: int, left: int, right: int
                               ) -> tuple[np.ndarray, torch.Tensor, str]:
         if video is None: