immich-app/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2025-12-17 09:13:17 +03:00
Code Issues Packages Projects Releases 16 Wiki Activity

fix(ml): minScore not being set correctly (#3916)

Browse Source 
* fixed `minScore` not being set correctly

* apply to init

* don't send `enabled`

* fix eslint warning

* better error message
This commit is contained in:
Mert
2023-08-30 04:16:00 -04:00
committed by GitHub
parent 343d89c032
commit df26e12db6
3 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
machine-learning/app/models/image_classification.py
View File

@@ -22,7 +22,7 @@ class ImageClassifier(InferenceModel):
cache_dir: Path | str | None = None,
**model_kwargs: Any,
) -> None:
self.min_score = min_score
self.min_score = model_kwargs.pop("minScore", min_score)
super().__init__(model_name, cache_dir, **model_kwargs)
def _download(self, **model_kwargs: Any) -> None:
@@ -65,4 +65,4 @@ class ImageClassifier(InferenceModel):
return tags
def configure(self, **model_kwargs: Any) -> None:
self.min_score = model_kwargs.get("min_score", self.min_score)
self.min_score = model_kwargs.pop("minScore", self.min_score)