From 2ad813c57d5c7c840d9c15ddc828cfcc209b7c23 Mon Sep 17 00:00:00 2001 From: Yasutaka Odo Date: Wed, 21 Dec 2022 01:03:52 +0900 Subject: [PATCH] undo unnecessary formatting --- whisperx/transcribe.py | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/whisperx/transcribe.py b/whisperx/transcribe.py index d02b3f7..054f2fb 100644 --- a/whisperx/transcribe.py +++ b/whisperx/transcribe.py @@ -1,21 +1,18 @@ import argparse import os 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 torch import torchaudio -import tqdm from transformers import AutoProcessor, Wav2Vec2ForCTC - -from .alignment import backtrack, get_trellis, merge_repeats, merge_words -from .audio import (HOP_LENGTH, N_FRAMES, SAMPLE_RATE, load_audio, - log_mel_spectrogram, pad_or_trim) +import tqdm +from .audio import SAMPLE_RATE, N_FRAMES, HOP_LENGTH, pad_or_trim, log_mel_spectrogram, load_audio +from .alignment import get_trellis, backtrack, merge_repeats, merge_words from .decoding import DecodingOptions, DecodingResult from .tokenizer import LANGUAGES, TO_LANGUAGE_CODE, get_tokenizer -from .utils import (exact_div, format_timestamp, optional_float, optional_int, - str2bool, write_ass, write_srt, write_txt, write_vtt) +from .utils import exact_div, format_timestamp, optional_int, optional_float, str2bool, write_txt, write_vtt, write_srt, write_ass if TYPE_CHECKING: from .model import Whisper