WAVbot commited on
Commit
8f04882
·
verified ·
1 Parent(s): aa4f829

Update inference_speech.py

Browse files
Files changed (1) hide show
  1. inference_speech.py +2 -2
inference_speech.py CHANGED
@@ -9,7 +9,7 @@ import torchaudio.transforms as T # Added for resampling
9
  # audio path
10
  parser = argparse.ArgumentParser()
11
  # --- Argument Parsing ---
12
- parser = argparse.ArgumentParser(description="Separate speech sources using Look2Hear TIGER model.")
13
  parser.add_argument("--audio_path", default="test/mix.wav", help="Path to audio file (mixture).")
14
  parser.add_argument("--output_dir", default="separated_audio", help="Directory to save separated audio files.")
15
  parser.add_argument("--model_cache_dir", default="cache", help="Directory to cache downloaded model.")
@@ -28,7 +28,7 @@ print(f"Using device: {device}")
28
 
29
  # Load model
30
 
31
- print("Loading TIGER model...")
32
  # Ensure cache directory exists if specified
33
  if cache_dir:
34
  os.makedirs(cache_dir, exist_ok=True)
 
9
  # audio path
10
  parser = argparse.ArgumentParser()
11
  # --- Argument Parsing ---
12
+ parser = argparse.ArgumentParser(description="Split audio with WAVbot.com")
13
  parser.add_argument("--audio_path", default="test/mix.wav", help="Path to audio file (mixture).")
14
  parser.add_argument("--output_dir", default="separated_audio", help="Directory to save separated audio files.")
15
  parser.add_argument("--model_cache_dir", default="cache", help="Directory to cache downloaded model.")
 
28
 
29
  # Load model
30
 
31
+ print("Loading WAvbot model...")
32
  # Ensure cache directory exists if specified
33
  if cache_dir:
34
  os.makedirs(cache_dir, exist_ok=True)