Update main.py

main.py

@@ -124,12 +124,12 @@ async def api_generate(request: Request, inputs: ApiInput):
     # Load person and fabric images from base64
     person = load_image_from_base64(inputs.person_base64)
     fabric = load_image_from_base64(inputs.fabric_base64)
-    if inputs.mask_base64 and inputs.mask_base64 != "" :
-        mask = load_image_from_base64(inputs.mask_base64, mode='L')
-        print( "Load image from base64 for mask image ..." );
-    else:
-        mask = generate_ultimate_mask(person_resized)
-        print( "Generating the mask image ..." );
+    # if inputs.mask_base64 and inputs.mask_base64 != "" :
+    #     mask = load_image_from_base64(inputs.mask_base64, mode='L')
+    #     print( "Load image from base64 for mask image ..." );
+    # else:
+    #     mask = generate_ultimate_mask(person_resized)
+    #     print( "Generating the mask image ..." );
     
     if person is None or fabric is None:
         raise HTTPException(status_code=400, detail="Could not decode base64 images.")
@@ -139,7 +139,7 @@ async def api_generate(request: Request, inputs: ApiInput):
     person_resized = person.resize(TARGET_SIZE, Image.Resampling.LANCZOS)
     
     # Handle mask image if provided
-    if inputs.mask_base64:
+    if inputs.mask_base64 and inputs.mask_base64 != "":
         mask = load_image_from_base64(inputs.mask_base64, mode='L')
         if mask is None:
             raise HTTPException(status_code=400, detail="Could not decode mask base64.")