Add Segment model

jhj0517 · Oct 29, 2024 · 95073dd · 95073dd
1 parent 2d14a2b
commit 95073dd
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/modules/whisper/data_classes.py b/modules/whisper/data_classes.py
@@ -16,6 +16,15 @@ class WhisperImpl(Enum):
     INSANELY_FAST_WHISPER = "insanely_fast_whisper"
 
 
+class Segment(BaseModel):
+    text: Optional[str] = Field(default=None,
+                                description="Transcription text of the segment")
+    start: Optional[float] = Field(default=None,
+                                   description="Start time of the segment")
+    end: Optional[float] = Field(default=None,
+                                 description="End time of the segment")
+
+
 class BaseParams(BaseModel):
     model_config = ConfigDict(protected_namespaces=())