Spaces:
Running
Running
jhj0517
commited on
Commit
·
a377305
1
Parent(s):
60840eb
Fix default value of the words
Browse files
modules/whisper/data_classes.py
CHANGED
@@ -19,7 +19,8 @@ class WhisperImpl(Enum):
|
|
19 |
|
20 |
|
21 |
class Segment(BaseModel):
|
22 |
-
id: Optional[
|
|
|
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 |
|