Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import asyncio
 import requests
 from flask import Flask, request, jsonify,send_file
 from PIL import Image
@@ -132,7 +133,8 @@ def save_image(img):
     return unique_name
 
 @spaces.GPU
-def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denoise_steps, seed, categorie = 'upper_body'):
+# Assurez-vous que les fonctions de traitement utilisent asyncio
+async def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denoise_steps, seed, categorie='upper_body'):
     device = "cuda"
     openpose_model.preprocessor.body_estimation.model.to(device)
     pipe.to(device)
@@ -156,20 +158,27 @@ def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denois
         human_img = human_img_orig.resize((768, 1024))
 
     if is_checked:
-        keypoints = openpose_model(human_img.resize((384, 512)))
-        model_parse, _ = parsing_model(human_img.resize((384, 512)))
-        mask, mask_gray = get_mask_location('hd', categorie , model_parse, keypoints)
+        # Utilisation de asyncio pour rendre le calcul non bloquant
+        keypoints = await asyncio.to_thread(openpose_model, human_img.resize((384, 512)))
+        model_parse, _ = await asyncio.to_thread(parsing_model, human_img.resize((384, 512)))
+        mask, mask_gray = await asyncio.to_thread(get_mask_location, 'hd', categorie, model_parse, keypoints)
         mask = mask.resize((768, 1024))
     else:
-        mask = dict['layers'][0].convert("RGB").resize((768, 1024))#pil_to_binary_mask(dict['layers'][0].convert("RGB").resize((768, 1024)))
+        mask = dict['layers'][0].convert("RGB").resize((768, 1024))
+
     mask_gray = (1 - transforms.ToTensor()(mask)) * tensor_transfrom(human_img)
     mask_gray = to_pil_image((mask_gray + 1.0) / 2.0)
 
     human_img_arg = _apply_exif_orientation(human_img.resize((384, 512)))
     human_img_arg = convert_PIL_to_numpy(human_img_arg, format="BGR")
 
-    args = apply_net.create_argument_parser().parse_args(('show', './configs/densepose_rcnn_R_50_FPN_s1x.yaml', './ckpt/densepose/model_final_162be9.pkl', 'dp_segm', '-v', '--opts', 'MODEL.DEVICE', 'cuda'))
-    pose_img = args.func(args, human_img_arg)
+    # Création des arguments de façon non bloquante
+    args = await asyncio.to_thread(
+        apply_net.create_argument_parser().parse_args,
+        ('show', './configs/densepose_rcnn_R_50_FPN_s1x.yaml', './ckpt/densepose/model_final_162be9.pkl', 'dp_segm', '-v', '--opts', 'MODEL.DEVICE', 'cuda')
+    )
+
+    pose_img = await asyncio.to_thread(args.func, args, human_img_arg)
     pose_img = pose_img[:, :, ::-1]
     pose_img = Image.fromarray(pose_img).resize((768, 1024))
 
@@ -212,7 +221,10 @@ def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denois
                 pose_img = tensor_transfrom(pose_img).unsqueeze(0).to(device, torch.float16)
                 garm_tensor = tensor_transfrom(garm_img).unsqueeze(0).to(device, torch.float16)
                 generator = torch.Generator(device).manual_seed(seed) if seed is not None else None
-                images = pipe(
+                
+                # L'appel principal au modèle doit également être asynchrone
+                images = await asyncio.to_thread(
+                    pipe,
                     prompt_embeds=prompt_embeds.to(device, torch.float16),
                     negative_prompt_embeds=negative_prompt_embeds.to(device, torch.float16),
                     pooled_prompt_embeds=pooled_prompt_embeds.to(device, torch.float16),
@@ -236,13 +248,12 @@ def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denois
         human_img_orig.paste(out_img, (int(left), int(top)))
         return human_img_orig, mask_gray
     else:
-        return images[0], mask_gray , mask
+        return images[0], mask_gray, mask
 
-        
-@app.route('/tryon-v2', methods=['POST'])
-def tryon_v2():
 
-    data = request.json
+@app.route('/tryon-v2', methods=['POST'])
+async def tryon_v2():
+    data = await request.json
     human_image_data = data['human_image']
     garment_image_data = data['garment_image']
 
@@ -267,11 +278,13 @@ def tryon_v2():
         'composite': None
     }
 
-    output_image, mask_image , mask = start_tryon(human_dict, garment_image, description, use_auto_mask, use_auto_crop, denoise_steps, seed, categorie)
+    # Exécuter le traitement principal de manière asynchrone
+    output_image, mask_image, mask = await start_tryon(human_dict, garment_image, description, use_auto_mask, use_auto_crop, denoise_steps, seed, categorie)
+    
     return jsonify({
         'image_id': save_image(output_image),
-        'mask_gray_id' : save_image(mask_image),
-        'mask_id' : save_image(mask)
+        'mask_gray_id': save_image(mask_image),
+        'mask_id': save_image(mask)
     })
 
 def clear_gpu_memory():