remove exir.capture from models.py (#2325)

Summary:
Pull Request resolved: https://github.com/pytorch/executorch/pull/2325

capture is deprecated, use export

bypass-github-pytorch-ci-checks

Reviewed By: mergennachin

Differential Revision: D54695851

fbshipit-source-id: fd23fce13a29cb76c8964ab009e8c72eef48e5e0