Update app.py
Browse files
app.py
CHANGED
@@ -123,7 +123,7 @@ EMPTY_CACHE = lambda: torch.cuda.empty_cache() if device == "cuda" else None
|
|
123 |
|
124 |
## ------------------------------ LOAD MODELS ------------------------------
|
125 |
|
126 |
-
|
127 |
def load_face_analyser_model(name="buffalo_l"):
|
128 |
global FACE_ANALYSER
|
129 |
if FACE_ANALYSER is None:
|
@@ -133,7 +133,7 @@ def load_face_analyser_model(name="buffalo_l"):
|
|
133 |
)
|
134 |
|
135 |
|
136 |
-
|
137 |
def load_face_swapper_model(path="./assets/pretrained_models/inswapper_128.onnx"):
|
138 |
global FACE_SWAPPER
|
139 |
if FACE_SWAPPER is None:
|
@@ -141,13 +141,13 @@ def load_face_swapper_model(path="./assets/pretrained_models/inswapper_128.onnx"
|
|
141 |
FACE_SWAPPER = Inswapper(model_file=path, batch_size=batch, providers=["CUDAExecutionProvider", "CPUExecutionProvider"])
|
142 |
|
143 |
|
144 |
-
|
145 |
def load_face_parser_model(path="./assets/pretrained_models/79999_iter.pth"):
|
146 |
global FACE_PARSER
|
147 |
if FACE_PARSER is None:
|
148 |
FACE_PARSER = init_parsing_model(path, device="cuda")
|
149 |
|
150 |
-
|
151 |
def load_nsfw_detector_model(path="./assets/pretrained_models/open-nsfw.onnx"):
|
152 |
global NSFW_DETECTOR
|
153 |
if NSFW_DETECTOR is None:
|
|
|
123 |
|
124 |
## ------------------------------ LOAD MODELS ------------------------------
|
125 |
|
126 |
+
|
127 |
def load_face_analyser_model(name="buffalo_l"):
|
128 |
global FACE_ANALYSER
|
129 |
if FACE_ANALYSER is None:
|
|
|
133 |
)
|
134 |
|
135 |
|
136 |
+
|
137 |
def load_face_swapper_model(path="./assets/pretrained_models/inswapper_128.onnx"):
|
138 |
global FACE_SWAPPER
|
139 |
if FACE_SWAPPER is None:
|
|
|
141 |
FACE_SWAPPER = Inswapper(model_file=path, batch_size=batch, providers=["CUDAExecutionProvider", "CPUExecutionProvider"])
|
142 |
|
143 |
|
144 |
+
|
145 |
def load_face_parser_model(path="./assets/pretrained_models/79999_iter.pth"):
|
146 |
global FACE_PARSER
|
147 |
if FACE_PARSER is None:
|
148 |
FACE_PARSER = init_parsing_model(path, device="cuda")
|
149 |
|
150 |
+
|
151 |
def load_nsfw_detector_model(path="./assets/pretrained_models/open-nsfw.onnx"):
|
152 |
global NSFW_DETECTOR
|
153 |
if NSFW_DETECTOR is None:
|