fix: configure docker build to use WHISPER_MODEL env

This commit is contained in:
Felix Spöttel
2023-06-28 11:07:23 +02:00
parent c41d3c6dd1
commit 136852afda
2 changed files with 2 additions and 5 deletions

View File

@@ -5,5 +5,3 @@ from whisper import _download, _MODELS # type: ignore
if __name__ == "__main__":
model_name = sys.argv[1].strip()
_download(_MODELS[model_name], "/models/", False)
# if model_name != "large":
# _download(_MODELS[f"{model_name}.en"], "/models/", False)