9 Commits

Author	SHA1	Message	Date
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
Mert	848ba685eb	fix(ml): race condition when loading models (#3207) ... * sync model loading, disabled model ttl by default * disable revalidation if model unloading disabled * moved lock	2023-07-11 12:01:21 -05:00
Mert	df1e8679d9	chore(ml): added testing and github workflow (#2969) ... * added testing * github action for python, made mypy happy * formatted with black * minor fixes and styling * test model cache * cache test dependencies * narrowed model cache tests * moved endpoint tests to their own class * cleaned up fixtures * formatting * removed unused dep	2023-06-27 18:21:33 -05:00
Mert	47982641b2	fix(ml): clear model cache on load error (#2951) ... * clear model cache on load error * updated caught exceptions	2023-06-27 16:01:24 -05:00
Mert	4d3ce0a65e	fixed setting different clip, removed unused stubs (#2987)	2023-06-27 12:21:50 -05:00
Mert	a2f5674bbb	refactor(ml): modularization and styling (#2835) ... * basic refactor and styling * removed batching * module entrypoint * removed unused imports * model superclass, model cache now in app state * fixed cache dir and enforced abstract method --------- Co-authored-by: Alex Tran <alex.tran1502@gmail.com>	2023-06-24 22:18:09 -05:00
Zeeshan Khan	34201be74c	feat(ml) backend takes image over HTTP (#2783) ... * using pydantic BaseSetting * ML API takes image file as input * keeping image in memory * reducing duplicate code * using bytes instead of UploadFile & other small code improvements * removed form-multipart, using HTTP body * format code --------- Co-authored-by: Alex Tran <alex.tran1502@gmail.com>	2023-06-17 22:49:19 -05:00
Mert	d0cc231782	feat(ml): model unloading (#2661) ... * model cache * fixed revalidation when using cache namespace * fixed ttl not being set, added lock	2023-06-06 20:48:51 -05:00
Mert	1e748864c5	chore(ml): updated dockerfile, added typing, packaging (#2642) ... * updated dockerfile, added typing, packaging apply env change * added arm64 support * added ml version pump, second try for arm64 * added linting config to pyproject.toml * renamed ml input field * fixed linter config * fixed dev docker compose	2023-06-05 09:40:48 -05:00