Charbel Malo commited on
Commit
044c785
·
verified ·
1 Parent(s): 838937a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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(duration=300, enable_queue=True)
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(duration=300, 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,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(duration=300, 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(duration=300, 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,7 +137,7 @@ load_face_swapper_model()
137
  ## ------------------------------ MAIN PROCESS ------------------------------
138
 
139
 
140
- @spaces.GPU(duration=300, enable_queue=True)
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,