Fix key error model_name issue by upgrade sentence-transformers version

Refer to https://github.com/UKPLab/sentence-transformers/issues/1241
Also user verbose flag passed through function parameters in image_search
This commit is contained in:
Saba
2021-12-11 11:58:19 -05:00
parent ce7a751e6b
commit 9536358d34
2 changed files with 2 additions and 2 deletions

View File

@@ -6,7 +6,7 @@ dependencies:
- numpy=1.*
- pytorch=1.*
- transformers=4.*
- sentence-transformers=2.0.0
- sentence-transformers=2.1.0
- fastapi=0.*
- uvicorn=0.*
- pyyaml=5.*