handling input as image or dict containing image and mask

app.py

@@ -162,8 +162,10 @@ def start_tryon(human_img_dict,garm_img,garment_des, background_img, is_checked,
     pipe.unet_encoder.to(device)
     # pipe.garment_net.to(device)
 
-    human_img_orig = human_img_dict["background"].convert("RGB")   # ImageEditor
-    #human_img_orig = human_img_dict.convert("RGB")     # Image
+    if isinstance(human_img_dict, dict):
+        human_img_orig = human_img_dict["background"].convert("RGB")   # ImageEditor
+    else:
+        human_img_orig = human_img_dict.convert("RGB")     # Image
     
 
     """