hysts HF staff commited on
Commit
0aff08c
1 Parent(s): 5838d81
Files changed (1) hide show
  1. model.py +4 -1
model.py CHANGED
@@ -1,6 +1,7 @@
1
  from __future__ import annotations
2
 
3
  import os
 
4
  import pickle
5
  import sys
6
 
@@ -9,7 +10,9 @@ import torch
9
  import torch.nn as nn
10
  from huggingface_hub import hf_hub_download
11
 
12
- sys.path.insert(0, 'StyleGAN-Human')
 
 
13
 
14
  HF_TOKEN = os.environ['HF_TOKEN']
15
 
 
1
  from __future__ import annotations
2
 
3
  import os
4
+ import pathlib
5
  import pickle
6
  import sys
7
 
 
10
  import torch.nn as nn
11
  from huggingface_hub import hf_hub_download
12
 
13
+ app_dir = pathlib.Path(__file__).parent
14
+ submodule_dir = app_dir / 'StyleGAN-Human'
15
+ sys.path.insert(0, submodule_dir.as_posix())
16
 
17
  HF_TOKEN = os.environ['HF_TOKEN']
18