harishB97 commited on
Commit
18f229b
·
verified ·
1 Parent(s): bc335c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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]