Update app.py

app.py

@@ -141,7 +141,7 @@ def start_tryon(
         left = (width - target_width) / 2
         top = (height - target_height) / 2
         right = (width + target_width) / 2
-        bottom = (height + target_width) / 2
+        bottom = (height + target_height) / 2
         cropped_img = human_img_orig.crop((left, top, right, bottom))
         crop_size = cropped_img.size
         human_img = cropped_img.resize((768, 1024))
@@ -258,10 +258,10 @@ def start_tryon(
         output_images.value.append(new_image)
 
         # ギャラリーを更新しつつmasked_imgを保持
-        yield output_images.value, gr.update()
+        yield output_images.value, mask_gray  # mask_grayを再度渡す
 
     # 最終的な結果を返す
-    return output_images.value, None
+    return output_images.value, mask_gray
 
 
 garm_list = os.listdir(os.path.join(example_path, "cloth"))