| | |
|---|
| | | self.punc_kwargs = punc_kwargs |
|---|
| | | self.spk_model = spk_model |
|---|
| | | self.spk_kwargs = spk_kwargs |
|---|
| | | self.model_path = kwargs.get("model_path", "./") |
|---|
| | | self.model_path = kwargs.get("model_path") |
|---|
| | | |
|---|
| | | |
|---|
| | | |
|---|
| | | def build_model(self, **kwargs): |
|---|
| | |
|---|
| | | if self.punc_model is not None: |
|---|
| | | self.punc_kwargs.update(cfg) |
|---|
| | | punc_res = self.inference(result["text"], model=self.punc_model, kwargs=self.punc_kwargs, **cfg) |
|---|
| | | result["text_with_punc"] = punc_res[0]["text"] |
|---|
| | | result["text"] = punc_res[0]["text"] |
|---|
| | | |
|---|
| | | # speaker embedding cluster after resorted |
|---|
| | | if self.spk_model is not None: |
|---|
