Spaces:
Runtime error
Runtime error
Commit
β’
82cef28
1
Parent(s):
aa95332
Update app.py
Browse files
app.py
CHANGED
@@ -18,11 +18,11 @@ from options.swap_options import SwapOptions
|
|
18 |
opt = SwapOptions().parse()
|
19 |
token = os.environ['token']
|
20 |
|
21 |
-
|
22 |
|
23 |
-
|
24 |
|
25 |
-
RetinaFace = load_model("RetinaFace-Res50.h5",
|
26 |
custom_objects={"FPN": FPN,
|
27 |
"SSH": SSH,
|
28 |
"BboxHead": BboxHead,
|
@@ -30,12 +30,12 @@ RetinaFace = load_model("RetinaFace-Res50.h5",
|
|
30 |
"ClassHead": ClassHead}
|
31 |
)
|
32 |
|
33 |
-
|
34 |
-
ArcFace = load_model("ArcFace-Res50.h5")
|
35 |
-
ArcFaceE = load_model("ArcFacePerceptual-Res50.h5")
|
36 |
|
37 |
-
|
38 |
-
G = load_model("FaceDancer_config_c_HQ.h5", custom_objects={"AdaIN": AdaIN,
|
39 |
"AdaptiveAttention": AdaptiveAttention,
|
40 |
"InstanceNormalization": InstanceNormalization})
|
41 |
|
|
|
18 |
opt = SwapOptions().parse()
|
19 |
token = os.environ['token']
|
20 |
|
21 |
+
retina_repo = Repository(local_dir="retina_models", clone_from="Farazquraishi/pendora")
|
22 |
|
23 |
+
from retinaface.models import *
|
24 |
|
25 |
+
RetinaFace = load_model("retina_models/RetinaFace-Res50.h5",
|
26 |
custom_objects={"FPN": FPN,
|
27 |
"SSH": SSH,
|
28 |
"BboxHead": BboxHead,
|
|
|
30 |
"ClassHead": ClassHead}
|
31 |
)
|
32 |
|
33 |
+
arc_repo = Repository(local_dir="arcface_model", clone_from="Farazquraishi/pendora")
|
34 |
+
ArcFace = load_model("arcface_model/ArcFace-Res50.h5")
|
35 |
+
ArcFaceE = load_model("arcface_model/ArcFacePerceptual-Res50.h5")
|
36 |
|
37 |
+
g_repo = Repository(local_dir="g_model_c_hq", clone_from="Farazquraishi/pendora",use_auth_token=token)
|
38 |
+
G = load_model("g_model_c_hq/FaceDancer_config_c_HQ.h5", custom_objects={"AdaIN": AdaIN,
|
39 |
"AdaptiveAttention": AdaptiveAttention,
|
40 |
"InstanceNormalization": InstanceNormalization})
|
41 |
|