Set default setting for tryon

app.py

@@ -134,7 +134,14 @@ pipe = TryonPipeline.from_pretrained(
 pipe.unet_encoder = UNet_Encoder
 
 @spaces.GPU
-def start_tryon(dict,garm_img,garment_des,is_automaskchecked,is_checked_crop,denoise_steps,seed):
+# For simple API
+def quick_tryon(humanTarget_img,garm_img,garment_prompt):
+    denoise_steps = 30
+    seed = 42
+    return start_tryon(humanTarget_img, garm_img, garment_prompt, True, True, denoise_steps, seed)
+
+
+def start_tryon(humanTarget_img,garm_img,garment_des,is_automaskchecked,is_checked_crop,denoise_steps,seed):
     device = "cuda"
 
     openpose_model.preprocessor.body_estimation.model.to(device)
@@ -142,7 +149,7 @@ def start_tryon(dict,garm_img,garment_des,is_automaskchecked,is_checked_crop,den
     pipe.unet_encoder.to(device)
 
     garm_img= garm_img.convert("RGB").resize((768,1024))
-    human_img_orig = dict["background"].convert("RGB")
+    human_img_orig = humanTarget_img.convert("RGB")
     
     if is_checked_crop:
         width, height = human_img_orig.size
@@ -175,7 +182,8 @@ def start_tryon(dict,garm_img,garment_des,is_automaskchecked,is_checked_crop,den
         # #     mask, mask_gray = get_mask_location('hd', "dresses", model_parse, keypoints)
         # #     mask = mask.resize((768,1024))
         # else:
-        mask = pil_to_binary_mask(dict['layers'][0].convert("RGB").resize((768, 1024)))
+        mask_temp_img = Image()
+        mask = pil_to_binary_mask(mask_temp_img.convert("RGB").resize((768, 1024)))
             # mask = transforms.ToTensor()(mask)
             # mask = mask.unsqueeze(0)
     mask_gray = (1-transforms.ToTensor()(mask)) * tensor_transfrom(human_img)
@@ -302,7 +310,7 @@ async def vton_run(
     target_human = Image.open(io.BytesIO(await upload_human.read()))
     target_cloth = Image.open(io.BytesIO(await upload_cloth.read()))
 
-    results = start_tryon(target_human, target_cloth, input_prompt, is_automaskchecked, is_checked_crop, denoise_steps, seed)
+    results = quick_tryon(target_human, target_cloth, input_prompt)
     return results[0]