|
|
|
@ -11,7 +11,6 @@
|
|
|
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
|
# See the License for the specific language governing permissions and
|
|
|
|
|
# limitations under the License.
|
|
|
|
|
import argparse
|
|
|
|
|
import re
|
|
|
|
|
|
|
|
|
|
import paddle
|
|
|
|
@ -24,6 +23,8 @@ from paddlespeech.text.models.ernie_linear import ErnieLinear
|
|
|
|
|
DefinedClassifier = {
|
|
|
|
|
'ErnieLinear': ErnieLinear,
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class Rhy_predictor():
|
|
|
|
|
def __init__(self, model_path, config_path, punc_path):
|
|
|
|
|
with open(config_path) as f:
|
|
|
|
|