Spaces:
Running
Running
Update vtoonify_model.py
Browse files- vtoonify_model.py +1 -1
vtoonify_model.py
CHANGED
@@ -144,7 +144,7 @@ class Model():
|
|
144 |
#@torch.inference_mode()
|
145 |
def detect_and_align_image(self, frame_rgb: np.ndarray, top: int, bottom: int, left: int, right: int) -> tuple[np.ndarray, torch.Tensor, str]:
|
146 |
if frame_rgb is None:
|
147 |
-
|
148 |
|
149 |
# Convert RGB to BGR
|
150 |
frame_bgr = cv2.cvtColor(frame_rgb, cv2.COLOR_RGB2BGR)
|
|
|
144 |
#@torch.inference_mode()
|
145 |
def detect_and_align_image(self, frame_rgb: np.ndarray, top: int, bottom: int, left: int, right: int) -> tuple[np.ndarray, torch.Tensor, str]:
|
146 |
if frame_rgb is None:
|
147 |
+
return np.zeros((256, 256, 3), np.uint8), None, 'Error: fail to load the image.'
|
148 |
|
149 |
# Convert RGB to BGR
|
150 |
frame_bgr = cv2.cvtColor(frame_rgb, cv2.COLOR_RGB2BGR)
|