Charbel Malo
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -100,7 +100,7 @@ EMPTY_CACHE = lambda: torch.cuda.empty_cache() if device == "cuda" else None
|
|
100 |
|
101 |
## ------------------------------ LOAD MODELS ------------------------------
|
102 |
|
103 |
-
@spaces.GPU(
|
104 |
def load_face_analyser_model(name="buffalo_l"):
|
105 |
global FACE_ANALYSER
|
106 |
if FACE_ANALYSER is None:
|
@@ -110,7 +110,7 @@ def load_face_analyser_model(name="buffalo_l"):
|
|
110 |
)
|
111 |
|
112 |
|
113 |
-
@spaces.GPU(
|
114 |
def load_face_swapper_model(path="./assets/pretrained_models/inswapper_128.onnx"):
|
115 |
global FACE_SWAPPER
|
116 |
if FACE_SWAPPER is None:
|
@@ -118,13 +118,13 @@ def load_face_swapper_model(path="./assets/pretrained_models/inswapper_128.onnx"
|
|
118 |
FACE_SWAPPER = Inswapper(model_file=path, batch_size=batch, providers=["CUDAExecutionProvider", "CPUExecutionProvider"])
|
119 |
|
120 |
|
121 |
-
@spaces.GPU(
|
122 |
def load_face_parser_model(path="./assets/pretrained_models/79999_iter.pth"):
|
123 |
global FACE_PARSER
|
124 |
if FACE_PARSER is None:
|
125 |
FACE_PARSER = init_parsing_model(path, device=device)
|
126 |
|
127 |
-
@spaces.GPU(
|
128 |
def load_nsfw_detector_model(path="./assets/pretrained_models/open-nsfw.onnx"):
|
129 |
global NSFW_DETECTOR
|
130 |
if NSFW_DETECTOR is None:
|
@@ -137,7 +137,7 @@ load_face_swapper_model()
|
|
137 |
## ------------------------------ MAIN PROCESS ------------------------------
|
138 |
|
139 |
|
140 |
-
@spaces.GPU(
|
141 |
def process(
|
142 |
input_type,
|
143 |
image_path,
|
|
|
100 |
|
101 |
## ------------------------------ LOAD MODELS ------------------------------
|
102 |
|
103 |
+
@spaces.GPU(enable_queue=True)
|
104 |
def load_face_analyser_model(name="buffalo_l"):
|
105 |
global FACE_ANALYSER
|
106 |
if FACE_ANALYSER is None:
|
|
|
110 |
)
|
111 |
|
112 |
|
113 |
+
@spaces.GPU(enable_queue=True)
|
114 |
def load_face_swapper_model(path="./assets/pretrained_models/inswapper_128.onnx"):
|
115 |
global FACE_SWAPPER
|
116 |
if FACE_SWAPPER is None:
|
|
|
118 |
FACE_SWAPPER = Inswapper(model_file=path, batch_size=batch, providers=["CUDAExecutionProvider", "CPUExecutionProvider"])
|
119 |
|
120 |
|
121 |
+
@spaces.GPU(enable_queue=True)
|
122 |
def load_face_parser_model(path="./assets/pretrained_models/79999_iter.pth"):
|
123 |
global FACE_PARSER
|
124 |
if FACE_PARSER is None:
|
125 |
FACE_PARSER = init_parsing_model(path, device=device)
|
126 |
|
127 |
+
@spaces.GPU(enable_queue=True)
|
128 |
def load_nsfw_detector_model(path="./assets/pretrained_models/open-nsfw.onnx"):
|
129 |
global NSFW_DETECTOR
|
130 |
if NSFW_DETECTOR is None:
|
|
|
137 |
## ------------------------------ MAIN PROCESS ------------------------------
|
138 |
|
139 |
|
140 |
+
@spaces.GPU(enable_queue=True)
|
141 |
def process(
|
142 |
input_type,
|
143 |
image_path,
|