Update app.py

app.py

@@ -100,6 +100,7 @@ EMPTY_CACHE = lambda: torch.cuda.empty_cache() if device == "cuda" else None
 
 ## ------------------------------ LOAD MODELS ------------------------------
 
+@spaces.GPU(duration=300, enable_queue=True)
 def load_face_analyser_model(name="buffalo_l"):
     global FACE_ANALYSER
     if FACE_ANALYSER is None:
@@ -109,6 +110,7 @@ def load_face_analyser_model(name="buffalo_l"):
         )
 
 
+@spaces.GPU(duration=300, enable_queue=True)
 def load_face_swapper_model(path="./assets/pretrained_models/inswapper_128.onnx"):
     global FACE_SWAPPER
     if FACE_SWAPPER is None:
@@ -116,11 +118,13 @@ def load_face_swapper_model(path="./assets/pretrained_models/inswapper_128.onnx"
         FACE_SWAPPER = Inswapper(model_file=path, batch_size=batch, providers=["CUDAExecutionProvider", "CPUExecutionProvider"])
 
 
+@spaces.GPU(duration=300, enable_queue=True)
 def load_face_parser_model(path="./assets/pretrained_models/79999_iter.pth"):
     global FACE_PARSER
     if FACE_PARSER is None:
         FACE_PARSER = init_parsing_model(path, device=device)
 
+@spaces.GPU(duration=300, enable_queue=True)
 def load_nsfw_detector_model(path="./assets/pretrained_models/open-nsfw.onnx"):
     global NSFW_DETECTOR
     if NSFW_DETECTOR is None: