Files
gradio-webrtc/backend/fastrtc/pause_detection/protocol.py
Freddy Boulton cbbfa17679 Add Method for loading community Vad Models (#136)
* Add code

* add code
2025-03-07 16:27:18 -05:00

21 lines
420 B
Python

from typing import Any, Protocol, TypeAlias
import numpy as np
from numpy.typing import NDArray
from ..utils import AudioChunk
ModelOptions: TypeAlias = Any
class PauseDetectionModel(Protocol):
def vad(
self,
audio: tuple[int, NDArray[np.int16] | NDArray[np.float32]],
options: ModelOptions,
) -> tuple[float, list[AudioChunk]]: ...
def warmup(
self,
) -> None: ...