felixrosberg commited on
Commit
c46bfbb
β€’
1 Parent(s): 9adeced

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -10
app.py CHANGED
@@ -18,8 +18,7 @@ token = os.environ['model_fetch']
18
 
19
  opt = SwapOptions().parse()
20
 
21
- retina_repo = Repository(local_dir="retina_model", clone_from="felixrosberg/retinaface_resnet50",
22
- private=True, use_auth_token=token)
23
 
24
  from retina_model.models import *
25
 
@@ -30,25 +29,21 @@ RetinaFace = load_model("retina_model/retinaface_res50.h5",
30
  "LandmarkHead": LandmarkHead,
31
  "ClassHead": ClassHead})
32
 
33
- arc_repo = Repository(local_dir="arcface_model", clone_from="felixrosberg/arcface_tf",
34
- private=True, use_auth_token=token)
35
  ArcFace = load_model("arcface_model/arc_res50.h5")
36
  ArcFaceE = load_model("arcface_model/arc_res50e.h5")
37
 
38
- g_repo = Repository(local_dir="g_model_c_hq", clone_from="felixrosberg/affa_config_c_hq",
39
- private=True, use_auth_token=token)
40
  G = load_model("g_model_c_hq/generator_t_28.h5", custom_objects={"AdaIN": AdaIN,
41
  "AdaptiveAttention": AdaptiveAttention,
42
  "InstanceNormalization": InstanceNormalization})
43
 
44
- r_repo = Repository(local_dir="reconstruction_attack", clone_from="felixrosberg/reconstruction_attack",
45
- private=True, use_auth_token=token)
46
  R = load_model("reconstruction_attack/reconstructor_42.h5", custom_objects={"AdaIN": AdaIN,
47
  "AdaptiveAttention": AdaptiveAttention,
48
  "InstanceNormalization": InstanceNormalization})
49
 
50
- permuter_repo = Repository(local_dir="identity_permuter", clone_from="felixrosberg/identitypermuter",
51
- private=True, use_auth_token=token, git_user="felixrosberg")
52
 
53
  from identity_permuter.id_permuter import identity_permuter
54
 
 
18
 
19
  opt = SwapOptions().parse()
20
 
21
+ retina_repo = Repository(local_dir="retina_model", clone_from="felixrosberg/retinaface_resnet50", use_auth_token=token)
 
22
 
23
  from retina_model.models import *
24
 
 
29
  "LandmarkHead": LandmarkHead,
30
  "ClassHead": ClassHead})
31
 
32
+ arc_repo = Repository(local_dir="arcface_model", clone_from="felixrosberg/arcface_tf", use_auth_token=token)
 
33
  ArcFace = load_model("arcface_model/arc_res50.h5")
34
  ArcFaceE = load_model("arcface_model/arc_res50e.h5")
35
 
36
+ g_repo = Repository(local_dir="g_model_c_hq", clone_from="felixrosberg/affa_config_c_hq", use_auth_token=token)
 
37
  G = load_model("g_model_c_hq/generator_t_28.h5", custom_objects={"AdaIN": AdaIN,
38
  "AdaptiveAttention": AdaptiveAttention,
39
  "InstanceNormalization": InstanceNormalization})
40
 
41
+ r_repo = Repository(local_dir="reconstruction_attack", clone_from="felixrosberg/reconstruction_attack", use_auth_token=token)
 
42
  R = load_model("reconstruction_attack/reconstructor_42.h5", custom_objects={"AdaIN": AdaIN,
43
  "AdaptiveAttention": AdaptiveAttention,
44
  "InstanceNormalization": InstanceNormalization})
45
 
46
+ permuter_repo = Repository(local_dir="identity_permuter", clone_from="felixrosberg/identitypermuter", use_auth_token=token, git_user="felixrosberg")
 
47
 
48
  from identity_permuter.id_permuter import identity_permuter
49