sanket kheni
commited on
Commit
·
c7fdde6
1
Parent(s):
5d7487a
app.py
CHANGED
|
@@ -16,6 +16,7 @@ from options.swap_options import SwapOptions
|
|
| 16 |
# token = os.environ['model_fetch']
|
| 17 |
|
| 18 |
opt = SwapOptions().parse()
|
|
|
|
| 19 |
|
| 20 |
retina_repo = Repository(local_dir="retina_models", clone_from="felixrosberg/RetinaFace")
|
| 21 |
|
|
@@ -33,7 +34,7 @@ arc_repo = Repository(local_dir="arcface_model", clone_from="felixrosberg/ArcFac
|
|
| 33 |
ArcFace = load_model("arcface_model/ArcFace-Res50.h5")
|
| 34 |
ArcFaceE = load_model("arcface_model/ArcFacePerceptual-Res50.h5")
|
| 35 |
|
| 36 |
-
g_repo = Repository(local_dir="g_model_c_hq", clone_from="felixrosberg/FaceDancer")
|
| 37 |
G = load_model("g_model_c_hq/FaceDancer_config_c_HQ.h5", custom_objects={"AdaIN": AdaIN,
|
| 38 |
"AdaptiveAttention": AdaptiveAttention,
|
| 39 |
"InstanceNormalization": InstanceNormalization})
|
|
|
|
| 16 |
# token = os.environ['model_fetch']
|
| 17 |
|
| 18 |
opt = SwapOptions().parse()
|
| 19 |
+
token = os.environ['token']
|
| 20 |
|
| 21 |
retina_repo = Repository(local_dir="retina_models", clone_from="felixrosberg/RetinaFace")
|
| 22 |
|
|
|
|
| 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})
|