face-swap

Running on CPU Upgrade

yotamsapi commited on Mar 21, 2023

Commit

d9edcea

•

1 Parent(s): b858004

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 retina_models.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 retinaface.models import *
 RetinaFace = load_model("retina_models/RetinaFace-Res50.h5",
                         custom_objects={"FPN": FPN,