18 lines
433 B
Python
18 lines
433 B
Python
from transcription.engines.BaseEngine import BaseEngine
|
|
|
|
class BatchSTTEngine(BaseEngine):
|
|
def __init__(
|
|
self,
|
|
modelName: str,
|
|
language: str,
|
|
dType: torch.dtype,
|
|
device: str
|
|
) -> None:
|
|
self.modelName = modelName
|
|
self.device = device
|
|
self.language = language
|
|
self.dType = dType
|
|
|
|
@abstractmethod
|
|
def transcribeBatch(self) -> None:
|
|
pass |