Update app.py
Browse files
app.py
CHANGED
@@ -57,7 +57,7 @@ def get_tree(imgpath):
|
|
57 |
if os.path.isdir(os.path.join(imgpath, foldername)):
|
58 |
folderpath = os.path.join(imgpath, foldername)
|
59 |
node_name = foldername
|
60 |
-
child_names = [filename.split('.')[0].split('-')[0] for filename in os.listdir(folderpath)]
|
61 |
|
62 |
if node_name in name_to_node:
|
63 |
node = name_to_node[node_name]
|
|
|
57 |
if os.path.isdir(os.path.join(imgpath, foldername)):
|
58 |
folderpath = os.path.join(imgpath, foldername)
|
59 |
node_name = foldername
|
60 |
+
child_names = list(set([filename.split('.')[0].split('-')[0] for filename in os.listdir(folderpath)]))
|
61 |
|
62 |
if node_name in name_to_node:
|
63 |
node = name_to_node[node_name]
|