keeping pose height and width variable and making it int, upgrading gradio version

README.md

@@ -4,7 +4,7 @@ emoji: 👕👔👚
 colorFrom: yellow
 colorTo: red
 sdk: gradio
-sdk_version: 4.24.0
+sdk_version: 4.26.0
 app_file: app.py
 pinned: false
 license: cc-by-nc-sa-4.0

app.py

@@ -123,6 +123,8 @@ pipe.unet_encoder = UNet_Encoder
 
 WIDTH = 4160 # 768
 HEIGHT = 6240 # 1024
+POSE_WIDTH = int(WIDTH/2)
+POSE_HEIGHT = int(HEIGHT/2)
 
 @spaces.GPU
 def start_tryon(dict,garm_img,garment_des,is_checked,is_checked_crop,denoise_steps,seed):
@@ -151,8 +153,8 @@ def start_tryon(dict,garm_img,garment_des,is_checked,is_checked_crop,denoise_ste
 
 
     if is_checked:
-        keypoints = openpose_model(human_img.resize((WIDTH/2,HEIGHT/2)))
-        model_parse, _ = parsing_model(human_img.resize((WIDTH/2,HEIGHT/2)))
+        keypoints = openpose_model(human_img.resize((POSE_WIDTH, POSE_HEIGHT)))
+        model_parse, _ = parsing_model(human_img.resize((POSE_WIDTH, POSE_HEIGHT)))
         mask, mask_gray = get_mask_location('hd', "upper_body", model_parse, keypoints)
         mask = mask.resize((WIDTH,HEIGHT))
     else: