Yuekai Zhang
2023-03-06 80e6c258cf89b5f11f4e52a4cc5a9cf2e95aa7be
funasr/runtime/triton_gpu/model_repo_paraformer_large_offline/scoring/1/model.py
@@ -21,6 +21,7 @@
import json
import os
import yaml
class TritonPythonModel:
    """Your Python model must use the same class name. Every Python model
@@ -73,9 +74,9 @@
        """
        load lang_char.txt
        """
        with open(str(vocab_file), 'r') as f:
            token_list = [line.strip() for line in f]
        return token_list
        with open(str(vocab_file), 'rb') as f:
            config = yaml.load(f, Loader=yaml.Loader)
        return config['token_list']
    def execute(self, requests):
        """`execute` must be implemented in every Python model. `execute`