Update app.py

app.py

@@ -19,6 +19,7 @@ from preprocess.humanparsing.run_parsing import Parsing
 from preprocess.openpose.run_openpose import OpenPose
 from detectron2.data.detection_utils import convert_PIL_to_numpy,_apply_exif_orientation
 from torchvision.transforms.functional import to_pil_image
+from datasets import Dataset
 
 
 def pil_to_binary_mask(pil_image, threshold=0):
@@ -203,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(half_img,bottomwear_img,bottomwear_des,"lower_body",is_checked,is_checked_crop,denoise_steps,seed)
+        IMG_data = [{"image": half_img}]  
+        img_dataset = Dataset.from_dict(IMG_data)
+        return start_tryon(img_dataset,bottomwear_img,bottomwear_des,"lower_body",is_checked,is_checked_crop,denoise_steps,seed)
     
         
 garm_list = os.listdir(os.path.join(example_path,"cloth"))
@@ -223,7 +226,7 @@ for ex_human in human_list_path:
 ##default human
 
 
-image_blocks = gr.Blocks(theme="Nymbo/Alyx_Theme").queue()
+image_blocks = gr.Blocks(theme=gr.themes.Ocean()).queue()
 with image_blocks as demo:
     gr.HTML("<center><h1>Virtual Try-On</h1></center>")
     gr.HTML("<center><p>Upload an image of a person and images of the clothes✨</p></center>")
@@ -266,7 +269,7 @@ with image_blocks as demo:
 
 
     with gr.Column():
-        try_button = gr.Button(value="Try-on")
+        try_button = gr.Button(value="Try-on",variant='primary')
         with gr.Accordion(label="Advanced Settings", open=False):
             with gr.Row():
                 denoise_steps = gr.Number(label="Denoising Steps", minimum=20, maximum=40, value=30, step=1)