Spaces:
Running
Running
Update inference_speech.py
Browse files- 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="
|
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
|
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)
|