Update BEN2.py
Browse files
BEN2.py
CHANGED
|
@@ -1028,7 +1028,7 @@ class BEN_Base(nn.Module):
|
|
| 1028 |
original_image.putalpha(mask)
|
| 1029 |
# mask = Image.fromarray(alpha)
|
| 1030 |
|
| 1031 |
-
return
|
| 1032 |
|
| 1033 |
|
| 1034 |
else:
|
|
|
|
| 1028 |
original_image.putalpha(mask)
|
| 1029 |
# mask = Image.fromarray(alpha)
|
| 1030 |
|
| 1031 |
+
return original_image
|
| 1032 |
|
| 1033 |
|
| 1034 |
else:
|