adirik commited on
Commit
b6090a0
1 Parent(s): 6271523
Files changed (2) hide show
  1. app.py +1 -1
  2. assets/.DS_Store +0 -0
app.py CHANGED
@@ -39,7 +39,7 @@ direction_list = []
39
  directions = [f for f in os.listdir(direction_folder) if f.endswith(".npz")]
40
  for d in directions:
41
  with np.load(direction_folder + d) as data:
42
- dir_name = f.split(".npz")[0]
43
  direction_list.append(data["s"])
44
  direction_map[dir_name] = {"direction": data["s"], "stylegan_type": "FFHQ"}
45
 
 
39
  directions = [f for f in os.listdir(direction_folder) if f.endswith(".npz")]
40
  for d in directions:
41
  with np.load(direction_folder + d) as data:
42
+ dir_name = d.split(".npz")[0]
43
  direction_list.append(data["s"])
44
  direction_map[dir_name] = {"direction": data["s"], "stylegan_type": "FFHQ"}
45
 
assets/.DS_Store DELETED
Binary file (6.15 kB)