tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

clean

Browse Source
This commit is contained in:
litagin02
2024-03-01 19:39:37 +09:00
parent f3475287b2
commit beb6a641be
1 changed files with 0 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
transcribe.py
View File

@@ -61,13 +61,9 @@ if __name__ == "__main__":
logger.warning(f"Failed to load model, so use `auto` compute_type: {e}") logger.warning(f"Failed to load model, so use `auto` compute_type: {e}")
model = WhisperModel(args.model, device=device) model = WhisperModel(args.model, device=device)
# wav_files = [
# os.path.join(input_dir, f) for f in os.listdir(input_dir) if f.endswith(".wav")
# ]
wav_files = [f for f in input_dir.rglob("*.wav") if f.is_file()] wav_files = [f for f in input_dir.rglob("*.wav") if f.is_file()]
if output_file.exists(): if output_file.exists():
logger.warning(f"{output_file} exists, backing up to {output_file}.bak") logger.warning(f"{output_file} exists, backing up to {output_file}.bak")
# if os.path.exists(output_file + ".bak"):
backup_path = output_file.with_name(output_file.name + ".bak") backup_path = output_file.with_name(output_file.name + ".bak")
if backup_path.exists(): if backup_path.exists():
logger.warning(f"{output_file}.bak exists, deleting...") logger.warning(f"{output_file}.bak exists, deleting...")