jamiya / app /model.py
jameszokah's picture
Refactor model imports across the application: update import paths from 'app.models' to 'app.model' in multiple files, and remove the now redundant models.py file.
ac008cf
raw
history blame contribute delete
266 Bytes
from dataclasses import dataclass
from typing import List, Tuple
import torch
@dataclass
class Segment:
"""A segment of speech with text, speaker, and audio."""
speaker: int
text: str
# (num_samples,), sample_rate = 24_000
audio: torch.Tensor