Spaces:
Sleeping
Sleeping
fix bug
Browse files- app.py +1 -1
- 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 =
|
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)
|
|