face-swap

Runtime error

yotamsapi commited on Mar 20, 2023

Commit

b858004

•

1 Parent(s): 8547084

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,7 +20,7 @@ token = os.environ['token']
 retina_repo = Repository(local_dir="retina_models", clone_from="felixrosberg/RetinaFace")
-from retinaface.models import *
 RetinaFace = load_model("retina_models/RetinaFace-Res50.h5",
                         custom_objects={"FPN": FPN,

 retina_repo = Repository(local_dir="retina_models", clone_from="felixrosberg/RetinaFace")
+from retina_models.models import *
 RetinaFace = load_model("retina_models/RetinaFace-Res50.h5",
                         custom_objects={"FPN": FPN,