Update app.py

app.py

@@ -124,7 +124,7 @@ pipe = TryonPipeline.from_pretrained(
 pipe.unet_encoder = UNet_Encoder
 
 @spaces.GPU
-def start_tryon(person_img, pose_img, mask_img, cloth_img, garment_des, denoise_steps, seed):
+def start_tryon(person_img, mask_img, cloth_img, garment_des, denoise_steps, seed):
     # Assuming device is set up (e.g., "cuda" or "cpu")
     pipe.to(device)
     pipe.unet_encoder.to(device)
@@ -133,6 +133,7 @@ def start_tryon(person_img, pose_img, mask_img, cloth_img, garment_des, denoise_
     garm_img = cloth_img.convert("RGB").resize((768, 1024))
     human_img = person_img.convert("RGB").resize((768, 1024))
     mask = pil_to_binary_mask(mask_img.convert("RGB").resize((768, 1024)))
+    pose_img=Image.open("00006_00.jpg")
 
     # Prepare pose image (already uploaded)
     pose_img = pose_img.resize((768, 1024))
@@ -214,7 +215,6 @@ with image_blocks as demo:
     with gr.Row():
         with gr.Column():
             person_img = gr.Image(label='Person Image', sources='upload', type="pil")
-            pose_img = gr.Image(label='Pose Image', sources='upload', type="pil")
             mask_img = gr.Image(label='Mask Image', sources='upload', type="pil")
 
         with gr.Column():
@@ -229,6 +229,6 @@ with image_blocks as demo:
             image_out = gr.Image(label="Output Image", elem_id="output-img", show_share_button=False)
 
     try_button = gr.Button(value="Try-on")
-    try_button.click(fn=start_tryon, inputs=[person_img, pose_img, mask_img, cloth_img, garment_des, denoise_steps, seed], outputs=[image_out], api_name='tryon')
+    try_button.click(fn=start_tryon, inputs=[person_img, mask_img, cloth_img, garment_des, denoise_steps, seed], outputs=[image_out], api_name='tryon')
 
 image_blocks.launch()