Update app.py

app.py

@@ -152,10 +152,13 @@ def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denois
             mask, mask_gray = get_mask_location('hd', category, model_parse, keypoints)
             mask = mask.resize((768,1024))
         except NotImplementedError:
-            status_message = f"카테고리 {category}에 대한 자동 마스크 생성이 구현되지 않았습니다. 사용자가 그린 마스크를 사용합니다."
-            mask = pil_to_binary_mask(dict['layers'][0].convert("RGB").resize((768, 1024)))
+            status_message = f"카테고리 {category}에 대한 자동 마스크 생성이 구현되지 않았습니다. 기본 마스크를 사용합니다."
+            mask = Image.new('L', (768, 1024), 255)
     else:
-        mask = pil_to_binary_mask(dict['layers'][0].convert("RGB").resize((768, 1024)))
+        if dict['layers'] and dict['layers'][0]:
+            mask = pil_to_binary_mask(dict['layers'][0].convert("RGB").resize((768, 1024)))
+        else:
+            mask = Image.new('L', (768, 1024), 255)
     
     mask_gray = (1-transforms.ToTensor()(mask)) * tensor_transfrom(human_img)
     mask_gray = to_pil_image((mask_gray+1.0)/2.0)
@@ -225,7 +228,12 @@ def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denois
                         width=768,
                         ip_adapter_image = garm_img.resize((768,1024)),
                         guidance_scale=2.0,
-                    )[0]
+                    ).images
+
+    print(f"Mask shape: {mask.size}")
+    print(f"Human image shape: {human_img.size}")
+    print(f"Garment image shape: {garm_img.size}")
+    print(f"Output image shape: {images[0].size}")
 
     if is_checked_crop:
         out_img = images[0].resize(crop_size)
@@ -294,9 +302,11 @@ with image_blocks as demo:
     with gr.Column():
         status_message = gr.Textbox(label="상태", interactive=False)
 
+
+
     try_button.click(fn=start_tryon, 
                      inputs=[imgs, garm_img, prompt, is_checked, is_checked_crop, denoise_steps, seed, category], 
                      outputs=[image_out, masked_img, status_message], 
                      api_name='tryon')
 
-image_blocks.launch(auth=("gini","pick"))
+image_blocks.launch(auth=("gini","pick"))