felixrosberg commited on
Commit
9bff996
β€’
1 Parent(s): 0ac3ff9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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="g_model", clone_from="felixrosberg/faceswapmodel",
37
  private=True, use_auth_token=token)
38
- G = load_model("g_model/affa_f_demo.h5", custom_objects={"AdaIN": AdaIN,
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