Update app.py

app.py

@@ -122,7 +122,7 @@ pipe = TryonPipeline.from_pretrained(
 pipe.unet_encoder = UNet_Encoder
 
 
-@spaces.GPU(duration=110)  # 実行時間を110秒に設定
+@spaces.GPU(duration=90)  # 実行時間を90秒に設定
 def start_tryon(
     dict, garm_img, garment_des, is_checked, is_checked_crop, denoise_steps, seed, num_images
 ):
@@ -142,7 +142,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))
@@ -248,9 +248,9 @@ def start_tryon(
             out_img = images[0].resize(crop_size)
             human_img_copy = human_img_orig.copy()
             human_img_copy.paste(out_img, (int(left), int(top)))
-            yield human_img_copy, None
+            yield [human_img_copy], None
         else:
-            yield images[0], None
+            yield [images[0]], None
 
 
 garm_list = os.listdir(os.path.join(example_path, "cloth"))
@@ -305,7 +305,7 @@ with image_blocks as demo:
                 label="Masked image output", elem_id="masked-img", show_share_button=False
             )
         with gr.Column():
-            image_out = gr.Image(label="Output", elem_id="output-img", show_share_button=False)
+            image_gallery = gr.Gallery(label="Generated Images", elem_id="output-gallery", show_label=True)
 
     with gr.Column():
         try_button = gr.Button(value="Try-on")
@@ -331,7 +331,7 @@ with image_blocks as demo:
             seed,
             num_images,
         ],
-        outputs=[image_out, masked_img],
+        outputs=[image_gallery, masked_img],
         api_name='tryon',
     )