adirik commited on
Commit
fe02fc9
1 Parent(s): b6090a0
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ 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
 
46
 
 
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(d)
44
  direction_map[dir_name] = {"direction": data["s"], "stylegan_type": "FFHQ"}
45
 
46