Update app.py

app.py

@@ -168,17 +168,13 @@ def save_image(img):
     img.save(unique_name, format="WEBP", lossless=True) 
     return unique_name
 
-# Initialisation globale des modèles
-device = "cuda"
-
-# Charger les modèles une seule fois globalement
-openpose_model.preprocessor.body_estimation.model.to(device)
-pipe.to(device)
-pipe.unet_encoder.to(device)
-
 @spaces.GPU
 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)
+    pipe.unet_encoder.to(device)
+
     garm_img = garm_img.convert("RGB").resize((768, 1024))
     human_img_orig = dict["background"].convert("RGB")
 
@@ -414,4 +410,4 @@ def get_image(image_id):
         return jsonify({'error': 'Image not found'}), 404
 
 if __name__ == "__main__":
-    app.run(debug=False, host="0.0.0.0", port=7860)
+    app.run(debug=False, host="0.0.0.0", port=7860)