Update app.py

app.py

@@ -18,37 +18,38 @@ token = os.environ['model_fetch']
 
 opt = SwapOptions().parse()
 
-retina_repo = Repository(local_dir="retina_model", clone_from="felixrosberg/retinaface_resnet50", use_auth_token=token)
+retina_repo = Repository(local_dir="retina_models", clone_from="felixrosberg/RetinaFace")
 
 from retina_model.models import *
 
-RetinaFace = load_model("retina_model/retinaface_res50.h5",
+RetinaFace = load_model("retina_model/RetinaFace-Res50.h5",
                         custom_objects={"FPN": FPN,
                                         "SSH": SSH,
                                         "BboxHead": BboxHead,
                                         "LandmarkHead": LandmarkHead,
                                         "ClassHead": ClassHead})
 
-arc_repo = Repository(local_dir="arcface_model", clone_from="felixrosberg/arcface_tf", use_auth_token=token)
-ArcFace = load_model("arcface_model/arc_res50.h5")
-ArcFaceE = load_model("arcface_model/arc_res50e.h5")
+arc_repo = Repository(local_dir="arcface_model", clone_from="felixrosberg/ArcFace")
+ArcFace = load_model("arcface_model/ArcFace-Res50.h5")
+ArcFaceE = load_model("arcface_model/ArcFacePerceptual-Res50.h5")
 
-g_repo = Repository(local_dir="g_model_c_hq", clone_from="felixrosberg/affa_config_c_hq", use_auth_token=token)
-G = load_model("g_model_c_hq/generator_t_28.h5", custom_objects={"AdaIN": AdaIN,
+
+g_repo = Repository(local_dir="g_model_c_hq", clone_from="felixrosberg/FaceDancer",use_auth_token=token)
+G = load_model("g_model_c_hq/FaceDancer_config_c_HQ.h5", custom_objects={"AdaIN": AdaIN,
                                                                  "AdaptiveAttention": AdaptiveAttention,
                                                                  "InstanceNormalization": InstanceNormalization})
 
-r_repo = Repository(local_dir="reconstruction_attack", clone_from="felixrosberg/reconstruction_attack", use_auth_token=token)
-R = load_model("reconstruction_attack/reconstructor_42.h5", custom_objects={"AdaIN": AdaIN,
-                                                                            "AdaptiveAttention": AdaptiveAttention,
-                                                                            "InstanceNormalization": InstanceNormalization})
+# r_repo = Repository(local_dir="reconstruction_attack", clone_from="felixrosberg/reconstruction_attack", use_auth_token=token)
+# R = load_model("reconstruction_attack/reconstructor_42.h5", custom_objects={"AdaIN": AdaIN,
+#                                                                             "AdaptiveAttention": AdaptiveAttention,
+#                                                                             "InstanceNormalization": InstanceNormalization})
 
-permuter_repo = Repository(local_dir="identity_permuter", clone_from="felixrosberg/identitypermuter", use_auth_token=token, git_user="felixrosberg")
+# permuter_repo = Repository(local_dir="identity_permuter", clone_from="felixrosberg/identitypermuter", use_auth_token=token, git_user="felixrosberg")
 
-from identity_permuter.id_permuter import identity_permuter
+# from identity_permuter.id_permuter import identity_permuter
 
-IDP = identity_permuter(emb_size=32, min_arg=False)
-IDP.load_weights("identity_permuter/id_permuter.h5")
+# IDP = identity_permuter(emb_size=32, min_arg=False)
+# IDP.load_weights("identity_permuter/id_permuter.h5")
 
 blend_mask_base = np.zeros(shape=(256, 256, 1))
 blend_mask_base[80:244, 32:224] = 1