harishB97 commited on
Commit
d2a3523
·
verified ·
1 Parent(s): a023ee5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -63,7 +63,8 @@ def get_tree(imgpath):
63
  node = name_to_node[node_name]
64
  else:
65
  node = Node(node_name)
66
- name_to_node[node_name] = id_to_node[node_name] = node
 
67
 
68
  child_nodes = []
69
  for child_name in child_names:
@@ -71,7 +72,8 @@ def get_tree(imgpath):
71
  child_node = name_to_node[child_name]
72
  else:
73
  child_node = Node(child_name)
74
- name_to_node[child_name] = id_to_node[node_name] = child_node
 
75
  child_node.parent = node
76
  child_nodes.append(child_node)
77
 
 
63
  node = name_to_node[node_name]
64
  else:
65
  node = Node(node_name)
66
+ name_to_node[node_name] = node
67
+ id_to_node[node.id] = node
68
 
69
  child_nodes = []
70
  for child_name in child_names:
 
72
  child_node = name_to_node[child_name]
73
  else:
74
  child_node = Node(child_name)
75
+ name_to_node[child_name] = child_node
76
+ id_to_node[child_node.id] = child_node
77
  child_node.parent = node
78
  child_nodes.append(child_node)
79