'Digraph' object has no attribute '_repr_svg_' in network_graph.ipynb

[justinge]

1. modify
   add src. before .model otherwise fail to import bodypose_model and handpose_model
   from src.model import bodypose_model, handpose_model
   but when run

bodymodel = bodypose_model()
hl.build_graph(bodymodel, torch.zeros([1, 3, 368, 368]))

it occurs error:

AttributeError Traceback (most recent call last)

~/miniconda3/envs/yolov5/lib/python3.9/site-packages/IPython/core/formatters.py in call(self, obj)
343 method = get_real_method(obj, self.print_method)
344 if method is not None:
--> 345 return method()
346 return None
347 else:

~/miniconda3/envs/yolov5/lib/python3.9/site-packages/hiddenlayer/graph.py in repr_svg(self)
357 def repr_svg(self):
358 """Allows Jupyter notebook to render the graph automatically."""
--> 359 return self.build_dot().repr_svg()
360
361 def save(self, path, format="pdf"):

AttributeError: 'Digraph' object has no attribute 'repr_svg'

12
<hiddenlayer.graph.Graph at 0x7fe8c272be50>