Update modules/whisper/faster_whisper_inference.py

modules/whisper/faster_whisper_inference.py

@@ -13,11 +13,10 @@ import gradio as gr
 from argparse import Namespace
 
 from modules.utils.paths import (FASTER_WHISPER_MODELS_DIR, DIARIZATION_MODELS_DIR, UVR_MODELS_DIR, OUTPUT_DIR)
-from modules.whisper.data_classes import *
-from modules.whisper.base_transcription_pipeline import BaseTranscriptionPipeline
+from modules.whisper.whisper_parameter import *
+from modules.whisper.whisper_base import WhisperBase
 
-
-class FasterWhisperInference(BaseTranscriptionPipeline):
+class FasterWhisperInference(WhisperBase):
     def __init__(self,
                  model_dir: str = FASTER_WHISPER_MODELS_DIR,
                  diarization_model_dir: str = DIARIZATION_MODELS_DIR,
@@ -98,11 +97,11 @@ class FasterWhisperInference(BaseTranscriptionPipeline):
             language_detection_segments=params.language_detection_segments,
             prompt_reset_on_temperature=params.prompt_reset_on_temperature,
         )
-        progress(0, desc="Loading audio..")
+        progress(0, desc="Loading audio...")
 
         segments_result = []
         for segment in segments:
-            progress(segment.start / info.duration, desc="Transcribing..")
+            progress(segment.start / info.duration, desc="Transcribing...")
             segments_result.append(Segment.from_faster_whisper(segment))
 
         elapsed_time = time.time() - start_time
@@ -127,7 +126,7 @@ class FasterWhisperInference(BaseTranscriptionPipeline):
         progress: gr.Progress
             Indicator to show progress directly in gradio.
         """
-        progress(0, desc="Initializing Model..")
+        progress(0, desc="Initializing Model...")
 
         model_size_dirname = model_size.replace("/", "--") if "/" in model_size else model_size
         if model_size not in self.model_paths and model_size_dirname not in self.model_paths: