multimodalart HF staff commited on
Commit
d6331eb
1 Parent(s): 0b56c75

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -31,7 +31,10 @@ for entry in os.listdir(directory_path):
31
  # Print file name and its size in bytes
32
  print(f"{entry}: {size} bytes")
33
 
34
- from HairFastGAN.hair_swap import HairFast, get_parser
 
 
 
35
 
36
  hair_fast = HairFast(get_parser().parse_args([]))
37
 
 
31
  # Print file name and its size in bytes
32
  print(f"{entry}: {size} bytes")
33
 
34
+ hairfastgan_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'HairFastGAN')
35
+ print(hairfastgan_path)
36
+ sys.path.append(hairfastgan_path)
37
+ from hair_swap import HairFast, get_parser
38
 
39
  hair_fast = HairFast(get_parser().parse_args([]))
40