mirror of
https://github.com/shirayu/whispering.git
synced 2024-06-02 21:39:28 +00:00
better cmdline printing
This commit is contained in:
parent
c6eefc95bd
commit
a89e9d2ff6
|
@ -5,6 +5,8 @@ import whisper
|
|||
|
||||
download_path = str(Path("~/.cache/whisper").expanduser())
|
||||
|
||||
print(f"Downloading whisper models to {download_path}")
|
||||
|
||||
# get models to download from argparse
|
||||
parser = argparse.ArgumentParser()
|
||||
|
||||
|
@ -13,5 +15,8 @@ parser.add_argument("--models", nargs="+", default=[])
|
|||
|
||||
args = parser.parse_args()
|
||||
|
||||
print(f"Downloading models: {args.models}")
|
||||
|
||||
for m in args.models:
|
||||
print(f"Downloading model {m} to {download_path}")
|
||||
whisper._download(whisper._MODELS[m], download_path, False)
|
||||
|
|
Loading…
Reference in a new issue