Update app.py

app.py

@@ -327,7 +327,8 @@ def generate_mask(human_img, categorie='upper_body'):
         # Générer les points clés et le masque
         keypoints = openpose_model(human_img_resized)
         model_parse, _ = parsing_model(human_img_resized)
-        mask, _ = get_mask_location('hd', categorie, model_parse, keypoints)
+        mask, mask_gray = get_mask_location('hd', categorie , model_parse, keypoints)
+        mask = mask.resize((768, 1024))
 
         # Redimensionner le masque à la taille d'origine de l'image
         mask_resized = mask.resize(human_img.size)