victorisgeek commited on
Commit
da86e5c
·
verified ·
1 Parent(s): f6e07f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -25
app.py CHANGED
@@ -17,31 +17,6 @@ theme = gr.themes.Default(
17
  background_fill_secondary='#eaeaea'
18
  )
19
 
20
- def load_face_analyser_model(name="buffalo_l"):
21
- global FACE_ANALYSER
22
- if FACE_ANALYSER is None:
23
- FACE_ANALYSER = insightface.app.FaceAnalysis(name=name, providers=PROVIDER)
24
- FACE_ANALYSER.prepare(
25
- ctx_id=0, det_size=(DETECT_SIZE, DETECT_SIZE), det_thresh=DETECT_THRESH
26
- )
27
-
28
-
29
- def load_face_swapper_model(path="./assets/pretrained_models/inswapper_128.onnx"):
30
- global FACE_SWAPPER
31
- if FACE_SWAPPER is None:
32
- batch = int(BATCH_SIZE) if device == "cuda" else 1
33
- FACE_SWAPPER = Inswapper(model_file=path, batch_size=batch, providers=PROVIDER)
34
-
35
-
36
- def load_face_parser_model(path="./assets/pretrained_models/79999_iter.pth"):
37
- global FACE_PARSER
38
- if FACE_PARSER is None:
39
- FACE_PARSER = init_parsing_model(path, device=device)
40
-
41
-
42
- load_face_analyser_model()
43
- load_face_swapper_model()
44
-
45
  def add_bbox_padding(bbox, margin=5):
46
  return [
47
  bbox[0] - margin,
 
17
  background_fill_secondary='#eaeaea'
18
  )
19
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
20
  def add_bbox_padding(bbox, margin=5):
21
  return [
22
  bbox[0] - margin,