Spaces:
Configuration error
Configuration error
File size: 3,751 Bytes
db7bf9a dc4f25f f5b5ebf 313814b 2a79f48 dc4f25f 2a79f48 313814b db7bf9a 313814b 4bdd7f2 2a79f48 4bdd7f2 db7bf9a dc4f25f db7bf9a 313814b db7bf9a 313814b db7bf9a 313814b 2a79f48 4bdd7f2 e01d72d dc4f25f e01d72d 2a79f48 e01d72d 4bdd7f2 2a79f48 4bdd7f2 db7bf9a dc4f25f db7bf9a 2a79f48 db7bf9a f5b5ebf 5aa421e f5b5ebf 2afe55d 9eed954 |
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 |
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]
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 isinstance(segment.words, list) else []),
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),
)
@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",
},
]
},
)
|