One options is :

Image(graph.get_graph().draw_png())

before running above code :

!pip install pygraphviz

If you are using Colab :

!apt-get install -y graphviz libgraphviz-dev
!pip install pygraphviz