Spaces:
Running
Running
Update main.py
Browse files
main.py
CHANGED
|
@@ -124,7 +124,7 @@ async def api_generate(request: Request, inputs: ApiInput):
|
|
| 124 |
# Load person and fabric images from base64
|
| 125 |
person = load_image_from_base64(inputs.person_base64)
|
| 126 |
fabric = load_image_from_base64(inputs.fabric_base64)
|
| 127 |
-
if inputs.mask_base64
|
| 128 |
mask = load_image_from_base64(inputs.mask_base64, mode='L')
|
| 129 |
print( "Load image from base64 for mask image ..." );
|
| 130 |
else:
|
|
|
|
| 124 |
# Load person and fabric images from base64
|
| 125 |
person = load_image_from_base64(inputs.person_base64)
|
| 126 |
fabric = load_image_from_base64(inputs.fabric_base64)
|
| 127 |
+
if inputs.mask_base64 and inputs.mask_base64 != "" :
|
| 128 |
mask = load_image_from_base64(inputs.mask_base64, mode='L')
|
| 129 |
print( "Load image from base64 for mask image ..." );
|
| 130 |
else:
|