chrissewell/whisperX
1
0
Fork 0
mirror of https://github.com/m-bain/whisperX.git synced 2025-07-01 18:17:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

undo unnecessary formatting

Browse Source
This commit is contained in:
Yasutaka Odo
2022-12-21 01:03:52 +09:00
parent 080dc1b337
commit 2ad813c57d
1 changed files with 5 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
whisperx/transcribe.py
View File

@ -1,21 +1,18 @@
import argparse import argparse
import os import os
import warnings import warnings
from typing import TYPE_CHECKING, Iterator, List, Optional, Tuple, Union from typing import List, Optional, Tuple, Union, Iterator, TYPE_CHECKING
import numpy as np import numpy as np
import torch import torch
import torchaudio import torchaudio
import tqdm
from transformers import AutoProcessor, Wav2Vec2ForCTC from transformers import AutoProcessor, Wav2Vec2ForCTC
import tqdm
from .alignment import backtrack, get_trellis, merge_repeats, merge_words from .audio import SAMPLE_RATE, N_FRAMES, HOP_LENGTH, pad_or_trim, log_mel_spectrogram, load_audio
from .audio import (HOP_LENGTH, N_FRAMES, SAMPLE_RATE, load_audio, from .alignment import get_trellis, backtrack, merge_repeats, merge_words
log_mel_spectrogram, pad_or_trim)
from .decoding import DecodingOptions, DecodingResult from .decoding import DecodingOptions, DecodingResult
from .tokenizer import LANGUAGES, TO_LANGUAGE_CODE, get_tokenizer from .tokenizer import LANGUAGES, TO_LANGUAGE_CODE, get_tokenizer
from .utils import (exact_div, format_timestamp, optional_float, optional_int, from .utils import exact_div, format_timestamp, optional_int, optional_float, str2bool, write_txt, write_vtt, write_srt, write_ass
str2bool, write_ass, write_srt, write_txt, write_vtt)
if TYPE_CHECKING: if TYPE_CHECKING:
from .model import Whisper from .model import Whisper