mgbam commited on
Commit
567a164
·
verified ·
1 Parent(s): 467b75f

Update visualization.py

Browse files
Files changed (1) hide show
  1. visualization.py +2 -2
visualization.py CHANGED
@@ -29,6 +29,6 @@ def create_medical_graph(query: str, docs: list) -> str:
29
  net.add_node(term_id, label=term, color="green")
30
  net.add_edge(doc_id, term_id)
31
 
32
- # Generate the HTML content directly without writing to a temporary file.
33
- html_content = net.generate_html(notebook=False, height="600px", width="100%")
34
  return html_content
 
29
  net.add_node(term_id, label=term, color="green")
30
  net.add_edge(doc_id, term_id)
31
 
32
+ # Generate the HTML content directly without unsupported keyword arguments.
33
+ html_content = net.generate_html(notebook=False)
34
  return html_content