.

app.py

@@ -33,22 +33,21 @@ arc_repo = Repository(local_dir="arcface_model", clone_from="felixrosberg/ArcFac
 ArcFace = load_model("arcface_model/arc_res50.h5")
 ArcFaceE = load_model("arcface_model/arc_res50e.h5")
 
-g_repo = Repository(local_dir="g_model_c_hq", clone_from="felixrosberg/affa_config_c_hq",
-                    private=True, 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")
+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",
-                    private=True, 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",
+#                     private=True, 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",
-                           private=True, use_auth_token=token, git_user="felixrosberg")
+# permuter_repo = Repository(local_dir="identity_permuter", clone_from="felixrosberg/identitypermuter",
+#                            private=True, 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")