Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -116,7 +116,7 @@ def load_face_swapper_model(path="./assets/pretrained_models/inswapper_128.onnx"
|
|
116 |
def load_face_parser_model(path="./assets/pretrained_models/79999_iter.pth"):
|
117 |
global FACE_PARSER
|
118 |
if FACE_PARSER is None:
|
119 |
-
FACE_PARSER = init_parsing_model(path, device=
|
120 |
|
121 |
def load_nsfw_detector_model(path="./assets/pretrained_models/open-nsfw.onnx"):
|
122 |
global NSFW_DETECTOR
|
|
|
116 |
def load_face_parser_model(path="./assets/pretrained_models/79999_iter.pth"):
|
117 |
global FACE_PARSER
|
118 |
if FACE_PARSER is None:
|
119 |
+
FACE_PARSER = init_parsing_model(path, device="cpu")
|
120 |
|
121 |
def load_nsfw_detector_model(path="./assets/pretrained_models/open-nsfw.onnx"):
|
122 |
global NSFW_DETECTOR
|