charbel-malo commited on
Commit
f026784
·
verified ·
1 Parent(s): d54dd23

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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
- @spaces.GPU(enable_queue=True)
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
- @spaces.GPU(enable_queue=True)
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
- @spaces.GPU(enable_queue=True)
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
- @spaces.GPU(enable_queue=True)
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: