Spaces:
Runtime error
Runtime error
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)
|