Update app.py

app.py

@@ -12,7 +12,6 @@ from transformers import (
 )
 from diffusers import DDPMScheduler, AutoencoderKL
 from typing import List
-
 import torch
 import os
 from transformers import AutoTokenizer
@@ -142,7 +141,7 @@ def start_tryon(
         left = (width - target_width) / 2
         top = (height - target_height) / 2
         right = (width + target_width) / 2
-        bottom = (height + target_height) / 2
+        bottom = (height + target_width) / 2
         cropped_img = human_img_orig.crop((left, top, right, bottom))
         crop_size = cropped_img.size
         human_img = cropped_img.resize((768, 1024))
@@ -258,8 +257,8 @@ def start_tryon(
         # 画像を追加
         output_images.value.append(new_image)
 
-        # ギャラリーを更新
-        yield output_images.value, None
+        # ギャラリーを更新しつつmasked_imgを保持
+        yield output_images.value, gr.update()
 
     # 最終的な結果を返す
     return output_images.value, None
@@ -347,4 +346,4 @@ with image_blocks as demo:
         api_name='tryon',
     )
 
-image_blocks.launch(show_error=True)
+image_blocks.launch(show_error=True)