Mert c73832bd9c

refactor(ml): model downloading (#3545) ...

* download facial recognition models

* download hf models

* simplified logic

* updated `predict` for facial recognition

* ensure download method is called

* fixed repo_id for clip

* fixed download destination

* use st's own `snapshot_download`

* conditional download

* fixed predict method

* check if loaded

* minor fixes

* updated mypy overrides

* added pytest-mock

* updated tests

* updated lock

2023-08-05 21:45:13 -05:00

..

__init__.py

fixed setting different clip, removed unused stubs (#2987)

2023-06-27 12:21:50 -05:00

base.py

refactor(ml): model downloading (#3545)

2023-08-05 21:45:13 -05:00

cache.py

fix(ml): race condition when loading models (#3207)

2023-07-11 12:01:21 -05:00

clip.py

refactor(ml): model downloading (#3545)

2023-08-05 21:45:13 -05:00

facial_recognition.py

refactor(ml): model downloading (#3545)

2023-08-05 21:45:13 -05:00

image_classification.py

refactor(ml): model downloading (#3545)

2023-08-05 21:45:13 -05:00