.

app.py

@@ -18,11 +18,11 @@ from utils.utils import (estimate_norm, get_lm, inverse_estimate_norm,
 
 opt = SwapOptions().parse()
 
-retina_repo = Repository(local_dir="retina_model", clone_from="felixrosberg/RetinaFace")
+retina_repo = Repository(local_dir="retina_models", clone_from="felixrosberg/RetinaFace")
 
-from retina_model import *
+from retina_models import *
 
-RetinaFace = load_model("retina_model/retinaface_res50.h5",
+RetinaFace = load_model("retina_models/retinaface_res50.h5",
                         # custom_objects={"FPN": FPN,
                         #                 "SSH": SSH,
                         #                 "BboxHead": BboxHead,
@@ -30,9 +30,9 @@ RetinaFace = load_model("retina_model/retinaface_res50.h5",
                         #                 "ClassHead": ClassHead}
                         )
 
-arc_repo = Repository(local_dir="arcface_model", clone_from="felixrosberg/ArcFace")
-ArcFace = load_model("arcface_model/arc_res50.h5")
-ArcFaceE = load_model("arcface_model/arc_res50e.h5")
+arc_repo = Repository(local_dir="arcface_models", clone_from="felixrosberg/ArcFace")
+ArcFace = load_model("arcface_models/arc_res50.h5")
+ArcFaceE = load_model("arcface_models/arc_res50e.h5")
 
 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,