Spaces:
Configuration error
Configuration error
File size: 4,361 Bytes
db7bf9a dc4f25f f5b5ebf 313814b 2a79f48 dc4f25f 2a79f48 313814b db7bf9a 313814b 4bdd7f2 2a79f48 4bdd7f2 db7bf9a dc4f25f db7bf9a 313814b db7bf9a 313814b db7bf9a 313814b ed63b7d 2a79f48 4bdd7f2 e01d72d dc4f25f e01d72d ed63b7d 2a79f48 e01d72d 4bdd7f2 2a79f48 ed63b7d 4bdd7f2 db7bf9a dc4f25f db7bf9a 2a79f48 db7bf9a f5b5ebf 5aa421e f5b5ebf 2afe55d 9eed954 a5d2e48 ed63b7d a5d2e48 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 |
from __future__ import annotations
from typing import TYPE_CHECKING, Literal
from pydantic import BaseModel, ConfigDict, Field
from faster_whisper_server.core import Segment, Transcription, Word, segments_to_text
if TYPE_CHECKING:
from faster_whisper.transcribe import TranscriptionInfo
# https://platform.openai.com/docs/api-reference/audio/json-object
class TranscriptionJsonResponse(BaseModel):
text: str
@classmethod
def from_segments(cls, segments: list[Segment]) -> TranscriptionJsonResponse:
return cls(text=segments_to_text(segments))
@classmethod
def from_transcription(cls, transcription: Transcription) -> TranscriptionJsonResponse:
return cls(text=transcription.text)
# https://platform.openai.com/docs/api-reference/audio/verbose-json-object
class TranscriptionVerboseJsonResponse(BaseModel):
task: str = "transcribe"
language: str
duration: float
text: str
words: list[Word] | None
segments: list[Segment]
@classmethod
def from_segment(cls, segment: Segment, transcription_info: TranscriptionInfo) -> TranscriptionVerboseJsonResponse:
return cls(
language=transcription_info.language,
duration=segment.end - segment.start,
text=segment.text,
words=segment.words if transcription_info.transcription_options.word_timestamps else None,
segments=[segment],
)
@classmethod
def from_segments(
cls, segments: list[Segment], transcription_info: TranscriptionInfo
) -> TranscriptionVerboseJsonResponse:
return cls(
language=transcription_info.language,
duration=transcription_info.duration,
text=segments_to_text(segments),
segments=segments,
words=Word.from_segments(segments) if transcription_info.transcription_options.word_timestamps else None,
)
@classmethod
def from_transcription(cls, transcription: Transcription) -> TranscriptionVerboseJsonResponse:
return cls(
language="english", # FIX: hardcoded
duration=transcription.duration,
text=transcription.text,
words=transcription.words,
segments=[], # FIX: hardcoded
)
class ModelListResponse(BaseModel):
data: list[ModelObject]
object: Literal["list"] = "list"
class ModelObject(BaseModel):
id: str
"""The model identifier, which can be referenced in the API endpoints."""
created: int
"""The Unix timestamp (in seconds) when the model was created."""
object_: Literal["model"] = Field(serialization_alias="object")
"""The object type, which is always "model"."""
owned_by: str
"""The organization that owns the model."""
language: list[str] = Field(default_factory=list)
"""List of ISO 639-3 supported by the model. It's possible that the list will be empty. This field is not a part of the OpenAI API spec and is added for convenience.""" # noqa: E501
model_config = ConfigDict(
populate_by_name=True,
json_schema_extra={
"examples": [
{
"id": "Systran/faster-whisper-large-v3",
"created": 1700732060,
"object": "model",
"owned_by": "Systran",
},
{
"id": "Systran/faster-distil-whisper-large-v3",
"created": 1711378296,
"object": "model",
"owned_by": "Systran",
},
{
"id": "bofenghuang/whisper-large-v2-cv11-french-ct2",
"created": 1687968011,
"object": "model",
"owned_by": "bofenghuang",
},
]
},
)
TimestampGranularities = list[Literal["segment", "word"]]
DEFAULT_TIMESTAMP_GRANULARITIES: TimestampGranularities = ["segment"]
TIMESTAMP_GRANULARITIES_COMBINATIONS: list[TimestampGranularities] = [
[], # should be treated as ["segment"]. https://platform.openai.com/docs/api-reference/audio/createTranscription#audio-createtranscription-timestamp_granularities
["segment"],
["word"],
["word", "segment"],
["segment", "word"], # same as ["word", "segment"] but order is different
]
|