Spaces:
Sleeping
Sleeping
Update apply_mask.py
Browse files- apply_mask.py +1 -1
apply_mask.py
CHANGED
|
@@ -68,7 +68,7 @@ def apply_mask_and_crop(input_folder, mask, output_folder):
|
|
| 68 |
|
| 69 |
# Save cropped image
|
| 70 |
cv2.imwrite(output_path, cropped_img)
|
| 71 |
-
if i = 0:
|
| 72 |
preview_path = os.path.join(output_folder, f"preview_{filename}")
|
| 73 |
cv2.imwrite(preview_path, cropped_img)
|
| 74 |
return preview_path
|
|
|
|
| 68 |
|
| 69 |
# Save cropped image
|
| 70 |
cv2.imwrite(output_path, cropped_img)
|
| 71 |
+
if i == 0:
|
| 72 |
preview_path = os.path.join(output_folder, f"preview_{filename}")
|
| 73 |
cv2.imwrite(preview_path, cropped_img)
|
| 74 |
return preview_path
|