alfabill commited on
Commit
61998f4
·
1 Parent(s): f888fa5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=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