Spaces:

power2
/

powerswp

Runtime error

sanket kheni commited on Dec 16, 2022

Commit

2e5f44a

1 Parent(s): 7576d48

.

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,11 +17,11 @@ from options.swap_options import SwapOptions
 opt = SwapOptions().parse()
-retina_repo = Repository(local_dir="retina_model", clone_from="felixrosberg/RetinaFace")
 from retina_model.models import *
-RetinaFace = load_model("retina_model/RetinaFace-Res50.h5",
                         custom_objects={"FPN": FPN,
                                         "SSH": SSH,
                                         "BboxHead": BboxHead,

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