Update app.py

app.py

@@ -93,7 +93,7 @@ def start_tryon(img,garm_img,garment_des,cloth_type,is_checked,is_checked_crop,d
     pipe.unet_encoder.to(device)
 
     garm_img= garm_img.convert("RGB").resize((768,1024))
-    human_img_orig = img.copy() #dict["background"].convert("RGB")    
+    human_img_orig = img["background"].convert("RGB")   
     
     if is_checked_crop:
         width, height = human_img_orig.size
@@ -204,7 +204,9 @@ def main_(imgs,topwear_img,topwear_des,bottomwear_img,bottomwear_des,dress_img,d
         return start_tryon(imgs,bottomwear_img,bottomwear_des,"lower_body",is_checked,is_checked_crop,denoise_steps,seed)
     elif topwear_img!=None and bottomwear_img!=None:
         _, half_img, half_mask = start_tryon(imgs,topwear_img,topwear_des,"upper_body",is_checked,is_checked_crop,denoise_steps,seed)
-        return start_tryon(img_dataset,bottomwear_img,bottomwear_des,"lower_body",is_checked,is_checked_crop,denoise_steps,seed)
+        half_dict= {}
+        half_dict['background'],half_dict['layers'],half_dict['composite'] = half_img,None,None
+        return start_tryon(half_dict,bottomwear_img,bottomwear_des,"lower_body",is_checked,is_checked_crop,denoise_steps,seed)
     
         
 garm_list = os.listdir(os.path.join(example_path,"cloth"))
@@ -230,7 +232,7 @@ with image_blocks as demo:
     gr.HTML("<center><p>Upload an image of a person and images of the clothes✨</p></center>")
     with gr.Row():
         with gr.Column():
-            inp_img = gr.ImageEditor(sources='upload',type="pil",label='Human. Mask with pen or use auto-masking',image_mode='RGB',interactive=True)
+            inp_img = gr.ImageEditor(sources='upload', type="pil", label='Human. Mask with pen or use auto-masking', interactive=True)
             with gr.Row():
                 is_checked = gr.Checkbox(label="Yes", info="Use auto-generated mask (Takes 5 seconds)",value=True)
             with gr.Row():