VoxSum / src /server /models /summarization.py
Luigi's picture
Add speaker name detection feature
77e98bd
raw
history blame contribute delete
396 Bytes
from __future__ import annotations
from pydantic import BaseModel, Field
class SummaryRequest(BaseModel):
transcript: str = Field(..., min_length=1)
llm_model: str
prompt: str = Field("Summarize the transcript below.")
generate_title: bool = Field(default=True)
class SpeakerNameDetectionRequest(BaseModel):
utterances: list = Field(..., min_length=1)
llm_model: str