Spaces:
Runtime error
Runtime error
Commit
·
9bff996
1
Parent(s):
0ac3ff9
Update app.py
Browse files
app.py
CHANGED
@@ -33,9 +33,9 @@ arc_repo = Repository(local_dir="arcface_model", clone_from="felixrosberg/arcfac
|
|
33 |
private=True, use_auth_token=token)
|
34 |
ArcFace = load_model("arcface_model/arc_res50.h5")
|
35 |
|
36 |
-
g_repo = Repository(local_dir="
|
37 |
private=True, use_auth_token=token)
|
38 |
-
G = load_model("
|
39 |
"AdaptiveAttention": AdaptiveAttention,
|
40 |
"InstanceNormalization": InstanceNormalization})
|
41 |
|
|
|
33 |
private=True, use_auth_token=token)
|
34 |
ArcFace = load_model("arcface_model/arc_res50.h5")
|
35 |
|
36 |
+
g_repo = Repository(local_dir="g_model_c_hq", clone_from="felixrosberg/affa_config_c_hq",
|
37 |
private=True, use_auth_token=token)
|
38 |
+
G = load_model("g_model_c_hq/generator_t_28.h5", custom_objects={"AdaIN": AdaIN,
|
39 |
"AdaptiveAttention": AdaptiveAttention,
|
40 |
"InstanceNormalization": InstanceNormalization})
|
41 |
|