Spaces:
Running
Running
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 |
|