.

app.py

@@ -20,14 +20,15 @@ opt = SwapOptions().parse()
 
 retina_repo = Repository(local_dir="retina_model", clone_from="felixrosberg/RetinaFace")
 
-# from retina_model.models import *
+from retina_model import *
 
 RetinaFace = load_model("retina_model/retinaface_res50.h5",
-                        custom_objects={"FPN": FPN,
-                                        "SSH": SSH,
-                                        "BboxHead": BboxHead,
-                                        "LandmarkHead": LandmarkHead,
-                                        "ClassHead": ClassHead})
+                        # custom_objects={"FPN": FPN,
+                        #                 "SSH": SSH,
+                        #                 "BboxHead": BboxHead,
+                        #                 "LandmarkHead": LandmarkHead,
+                        #                 "ClassHead": ClassHead}
+                        )
 
 arc_repo = Repository(local_dir="arcface_model", clone_from="felixrosberg/ArcFace")
 ArcFace = load_model("arcface_model/arc_res50.h5")