add private module to create model object for speech api

This commit is contained in:
jordi.hasianta
2022-09-15 19:47:27 +07:00
parent aeb4a53a2f
commit 2ec99681f0
3 changed files with 51 additions and 0 deletions

View File

@@ -0,0 +1,16 @@
import tempfile
from opendbm.api_lib.model import OPENDBM_DATA, AudioModel
from opendbm.dbm_lib.dbm_features.raw_features.nlp.transcribe import run_transcribe
class Transcribe(AudioModel):
def __init__(self):
super().__init__()
self._params = ["nlp_transcribe", "nlp_totalTime"]
@AudioModel.prep_func
def _fit_transform(self, path):
return run_transcribe(
path, f"{tempfile.gettempdir()}/", self.r_config, OPENDBM_DATA, save=False
)