hysts HF staff commited on
Commit
58dfe83
1 Parent(s): e38549b
Files changed (1) hide show
  1. dualstylegan.py +3 -1
dualstylegan.py CHANGED
@@ -19,7 +19,9 @@ if os.environ.get('SYSTEM') == 'spaces':
19
  os.system("sed -i '10,17d' DualStyleGAN/model/stylegan/op/fused_act.py")
20
  os.system("sed -i '10,17d' DualStyleGAN/model/stylegan/op/upfirdn2d.py")
21
 
22
- sys.path.insert(0, 'DualStyleGAN')
 
 
23
 
24
  from model.dualstylegan import DualStyleGAN
25
  from model.encoder.align_all_parallel import align_face
 
19
  os.system("sed -i '10,17d' DualStyleGAN/model/stylegan/op/fused_act.py")
20
  os.system("sed -i '10,17d' DualStyleGAN/model/stylegan/op/upfirdn2d.py")
21
 
22
+ app_dir = pathlib.Path(__file__).parent
23
+ submodule_dir = app_dir / 'DualStyleGAN'
24
+ sys.path.insert(0, submodule_dir.as_posix())
25
 
26
  from model.dualstylegan import DualStyleGAN
27
  from model.encoder.align_all_parallel import align_face