Update app.py

app.py

@@ -66,6 +66,8 @@ UNet_Encoder = UNet2DConditionModel_ref.from_pretrained(base_path, subfolder="un
 parsing_model = Parsing(0)
 openpose_model = OpenPose(0)
 
+@spaces.GPU
+
 # Préparation du pipeline Tryon
 pipe = TryonPipeline.from_pretrained(
     base_path,
@@ -132,7 +134,6 @@ def save_image(img):
     img.save(unique_name, format="WEBP", lossless=True) 
     return unique_name
 
-@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)
@@ -339,7 +340,6 @@ def get_image(image_id):
 		
 		
 
-@spaces.GPU 
 def generate_mask(human_img, categorie='upper_body'):
     device = "cuda"
     openpose_model.preprocessor.body_estimation.model.to(device)