Add factory function

modules/whisper/data_classes.py

@@ -30,6 +30,22 @@ class Segment(BaseModel):
     no_speech_prob: Optional[float] = Field(default=None, description="Probability that it's not speech")
     words: Optional[List['Word']] = Field(default=None, description="List of words contained in the segment")
 
+    @classmethod
+    def from_faster_whisper(cls,
+                            seg: faster_whisper.transcribe.Segment):
+        return cls(
+            id=seg.id,
+            text=seg.text,
+            start=seg.start,
+            end=seg.end,
+            tokens=seg.tokens,
+            temperature=seg.temperature,
+            avg_logprob=seg.avg_logprob,
+            compression_ratio=seg.compression_ratio,
+            no_speech_prob=seg.no_speech_prob,
+            words=seg.words
+        )
+
 
 class Word(NamedTuple):
     start: Optional[float] = None

modules/whisper/faster_whisper_inference.py

@@ -102,11 +102,7 @@ class FasterWhisperInference(BaseTranscriptionPipeline):
         segments_result = []
         for segment in segments:
             progress(segment.start / info.duration, desc="Transcribing..")
-            segments_result.append(Segment(
-                start=segment.start,
-                end=segment.end,
-                text=segment.text
-            ))
+            segments_result.append(Segment.from_faster_whisper(segment))
 
         elapsed_time = time.time() - start_time
         return segments_result, elapsed_time