Add progress bar for cli inference

Fixes #99
sigsep · Jul 23, 2021 · 8c40c46 · 8c40c46
1 parent 0f174b7
commit 8c40c46
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/openunmix/cli.py b/openunmix/cli.py
@@ -3,7 +3,7 @@
 import torchaudio
 import json
 import numpy as np
-
+import tqdm
 
 from openunmix import utils
 from openunmix import predict
@@ -152,7 +152,7 @@ def separate():
             raise RuntimeError("Please install pip package `stempeg`")
 
     # loop over the files
-    for input_file in args.input:
+    for input_file in tqdm.tqdm(args.input):
         if args.audio_backend == "stempeg":
             audio, rate = stempeg.read_stems(
                 input_file,