Feat: cache_dir

[wheynelau]

Small QOL change, adding a cache_dir argument to the input of from_pretrained.

In addition, removes the model after test_model runs, this reduces clutter, but may slow down frequent testing. (This can be removed or modified)

Fixes #132