Update app.py
Browse files
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] =
|
|
|
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] =
|
|
|
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 |
|