adirik commited on
Commit
1ec482b
1 Parent(s): 7887125
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(d)
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(dir_name)
44
  direction_map[dir_name] = {"direction": data["s"], "stylegan_type": "FFHQ"}
45
 
46