Farazquraishi commited on
Commit
8acef0e
β€’
1 Parent(s): 82cef28

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -18,7 +18,7 @@ from options.swap_options import SwapOptions
18
  opt = SwapOptions().parse()
19
  token = os.environ['token']
20
 
21
- retina_repo = Repository(local_dir="retina_models", clone_from="Farazquraishi/pendora")
22
 
23
  from retinaface.models import *
24
 
@@ -30,11 +30,11 @@ RetinaFace = load_model("retina_models/RetinaFace-Res50.h5",
30
  "ClassHead": ClassHead}
31
  )
32
 
33
- arc_repo = Repository(local_dir="arcface_model", clone_from="Farazquraishi/pendora")
34
  ArcFace = load_model("arcface_model/ArcFace-Res50.h5")
35
  ArcFaceE = load_model("arcface_model/ArcFacePerceptual-Res50.h5")
36
 
37
- g_repo = Repository(local_dir="g_model_c_hq", clone_from="Farazquraishi/pendora",use_auth_token=token)
38
  G = load_model("g_model_c_hq/FaceDancer_config_c_HQ.h5", custom_objects={"AdaIN": AdaIN,
39
  "AdaptiveAttention": AdaptiveAttention,
40
  "InstanceNormalization": InstanceNormalization})
 
18
  opt = SwapOptions().parse()
19
  token = os.environ['token']
20
 
21
+ retina_repo = Repository(local_dir="retina_models", clone_from="felixrosberg/RetinaFace")
22
 
23
  from retinaface.models import *
24
 
 
30
  "ClassHead": ClassHead}
31
  )
32
 
33
+ arc_repo = Repository(local_dir="arcface_model", clone_from="felixrosberg/ArcFace")
34
  ArcFace = load_model("arcface_model/ArcFace-Res50.h5")
35
  ArcFaceE = load_model("arcface_model/ArcFacePerceptual-Res50.h5")
36
 
37
+ g_repo = Repository(local_dir="g_model_c_hq", clone_from="felixrosberg/FaceDancer",use_auth_token=token)
38
  G = load_model("g_model_c_hq/FaceDancer_config_c_HQ.h5", custom_objects={"AdaIN": AdaIN,
39
  "AdaptiveAttention": AdaptiveAttention,
40
  "InstanceNormalization": InstanceNormalization})