jhj0517 commited on
Commit
a377305
·
1 Parent(s): 60840eb

Fix default value of the words

Browse files
Files changed (1) hide show
  1. modules/whisper/data_classes.py +4 -2
modules/whisper/data_classes.py CHANGED
@@ -19,7 +19,8 @@ class WhisperImpl(Enum):
19
 
20
 
21
  class Segment(BaseModel):
22
- id: Optional[str] = Field(default=None, description="Unique identifier for the segment")
 
23
  text: Optional[str] = Field(default=None, description="Transcription text of the segment")
24
  start: Optional[float] = Field(default=None, description="Start time of the segment")
25
  end: Optional[float] = Field(default=None, description="End time of the segment")
@@ -35,6 +36,7 @@ class Segment(BaseModel):
35
  seg: faster_whisper.transcribe.Segment):
36
  return cls(
37
  id=seg.id,
 
38
  text=seg.text,
39
  start=seg.start,
40
  end=seg.end,
@@ -43,7 +45,7 @@ class Segment(BaseModel):
43
  avg_logprob=seg.avg_logprob,
44
  compression_ratio=seg.compression_ratio,
45
  no_speech_prob=seg.no_speech_prob,
46
- words=seg.words
47
  )
48
 
49
 
 
19
 
20
 
21
  class Segment(BaseModel):
22
+ id: Optional[int] = Field(default=None, description="Incremental id for the segment")
23
+ seek: Optional[int] = Field(default=None, description="Seek of the segment from chunked audio")
24
  text: Optional[str] = Field(default=None, description="Transcription text of the segment")
25
  start: Optional[float] = Field(default=None, description="Start time of the segment")
26
  end: Optional[float] = Field(default=None, description="End time of the segment")
 
36
  seg: faster_whisper.transcribe.Segment):
37
  return cls(
38
  id=seg.id,
39
+ seek=seg.seek,
40
  text=seg.text,
41
  start=seg.start,
42
  end=seg.end,
 
45
  avg_logprob=seg.avg_logprob,
46
  compression_ratio=seg.compression_ratio,
47
  no_speech_prob=seg.no_speech_prob,
48
+ words=[] if seg.words is None else seg.words
49
  )
50
 
51