demo.py (2 lines): - line 63: # TODO: replace p3d_hat with model preditions - line 74: # TODO: remove break