Spaces:
Runtime error
Runtime error
Commit
Β·
8acef0e
1
Parent(s):
82cef28
Update app.py
Browse files
app.py
CHANGED
@@ -18,7 +18,7 @@ from options.swap_options import SwapOptions
|
|
18 |
opt = SwapOptions().parse()
|
19 |
token = os.environ['token']
|
20 |
|
21 |
-
retina_repo = Repository(local_dir="retina_models", clone_from="
|
22 |
|
23 |
from retinaface.models import *
|
24 |
|
@@ -30,11 +30,11 @@ RetinaFace = load_model("retina_models/RetinaFace-Res50.h5",
|
|
30 |
"ClassHead": ClassHead}
|
31 |
)
|
32 |
|
33 |
-
arc_repo = Repository(local_dir="arcface_model", clone_from="
|
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="
|
38 |
G = load_model("g_model_c_hq/FaceDancer_config_c_HQ.h5", custom_objects={"AdaIN": AdaIN,
|
39 |
"AdaptiveAttention": AdaptiveAttention,
|
40 |
"InstanceNormalization": InstanceNormalization})
|
|
|
18 |
opt = SwapOptions().parse()
|
19 |
token = os.environ['token']
|
20 |
|
21 |
+
retina_repo = Repository(local_dir="retina_models", clone_from="felixrosberg/RetinaFace")
|
22 |
|
23 |
from retinaface.models import *
|
24 |
|
|
|
30 |
"ClassHead": ClassHead}
|
31 |
)
|
32 |
|
33 |
+
arc_repo = Repository(local_dir="arcface_model", clone_from="felixrosberg/ArcFace")
|
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="felixrosberg/FaceDancer",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})
|