Update app.py

app.py

@@ -5,7 +5,7 @@ import networkx as nx
 
 
 def render_dependency_tree(words, parents, labels):
-    fig, ax = plt.subplots(figsize=(10, 6))
+    fig, ax = plt.subplots(figsize=(32, 16))
 
     # Create a directed graph
     G = nx.DiGraph()
@@ -16,8 +16,8 @@ def render_dependency_tree(words, parents, labels):
 
     # Adding edges with labels
     for i, (parent, label) in enumerate(zip(parents, labels)):
-        if parent != -1:
-            G.add_edge(parent, i, label=label)
+        if parent != 0:
+            G.add_edge(parent - 1, i, label=label)
 
     # Position nodes using Graphviz
     pos = nx.nx_agraph.graphviz_layout(G, prog='dot')