davda54 commited on
Commit
a6377ae
·
1 Parent(s): d74c2a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -5,7 +5,7 @@ import networkx as nx
5
 
6
 
7
  def render_dependency_tree(words, parents, labels):
8
- fig, ax = plt.subplots(figsize=(32, 24))
9
 
10
  # Create a directed graph
11
  G = nx.DiGraph()
@@ -24,12 +24,12 @@ def render_dependency_tree(words, parents, labels):
24
 
25
  # Draw the graph
26
  nx.draw(G, pos, ax=ax, with_labels=True, labels=nx.get_node_attributes(G, 'label'),
27
- arrows=True, node_color='#eeeeee', node_size=1000, node_shape='o', font_size=10
28
  )
29
 
30
  # Draw edge labels
31
  edge_labels = nx.get_edge_attributes(G, 'label')
32
- nx.draw_networkx_edge_labels(G, pos, ax=ax, edge_labels=edge_labels, rotate=False, alpha=0.9, font_size=8)
33
 
34
  return fig
35
 
 
5
 
6
 
7
  def render_dependency_tree(words, parents, labels):
8
+ fig, ax = plt.subplots(figsize=(32, 18))
9
 
10
  # Create a directed graph
11
  G = nx.DiGraph()
 
24
 
25
  # Draw the graph
26
  nx.draw(G, pos, ax=ax, with_labels=True, labels=nx.get_node_attributes(G, 'label'),
27
+ arrows=True, node_color='#ffffff', node_size=1000, node_shape='o', font_size=16
28
  )
29
 
30
  # Draw edge labels
31
  edge_labels = nx.get_edge_attributes(G, 'label')
32
+ nx.draw_networkx_edge_labels(G, pos, ax=ax, edge_labels=edge_labels, rotate=False, alpha=0.9, font_size=12)
33
 
34
  return fig
35