'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 _reprsvg(self) 357 def _reprsvg(self): 358 """Allows Jupyter notebook to render the graph automatically.""" --> 359 return self.build_dot()._reprsvg() 360 361 def save(self, path, format="pdf"):

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

12 <hiddenlayer.graph.Graph at 0x7fe8c272be50>