changing arm width to dc, short sleeve

app.py

@@ -131,7 +131,7 @@ WIDTH = int(768)
 HEIGHT = int(1024)
 POSE_WIDTH = int(WIDTH/2)  # int(WIDTH/2)
 POSE_HEIGHT = int(HEIGHT/2)  #int(HEIGHT/2)
-
+ARM_WIDTH = "dc" # "hd" # hd -> full sleeve, dc for half sleeve 
 CATEGORY = "upper_body" # "lower_body"
 
 def is_cropping_required(width, height):
@@ -196,7 +196,7 @@ def start_tryon(human_img_dict,garm_img,garment_des, background_img, is_checked,
         keypoints = openpose_model(human_img.resize((POSE_WIDTH, POSE_HEIGHT)))
         model_parse, _ = parsing_model(human_img.resize((POSE_WIDTH, POSE_HEIGHT)))
         # internally get mask location function is resizing model_parse to 384x512 if width & height not passed
-        mask, mask_gray = get_mask_location('hd', CATEGORY, model_parse, keypoints)
+        mask, mask_gray = get_mask_location(ARM_WIDTH, CATEGORY, model_parse, keypoints)
         mask = mask.resize((WIDTH, HEIGHT))
         logging.info("Mask location on model identified")
     else: