From b7cb19b01a1454f7a1388e24dcd4e10fc654bd7c Mon Sep 17 00:00:00 2001
From: shixian.shi <shixian.shi@alibaba-inc.com>
Date: 星期二, 16 一月 2024 11:30:25 +0800
Subject: [PATCH] update demo, readme
---
examples/industrial_data_pretraining/ct_transformer_streaming/demo.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/examples/industrial_data_pretraining/ct_transformer_streaming/demo.py b/examples/industrial_data_pretraining/ct_transformer_streaming/demo.py
index 5ef8381..081fd19 100644
--- a/examples/industrial_data_pretraining/ct_transformer_streaming/demo.py
+++ b/examples/industrial_data_pretraining/ct_transformer_streaming/demo.py
@@ -12,7 +12,7 @@
rec_result_all = "outputs: "
cache = {}
for vad in vads:
- rec_result = model(input=vad, cache=cache)
+ rec_result = model.generate(input=vad, cache=cache)
print(rec_result)
rec_result_all += rec_result[0]['text']
--
Gitblit v1.9.1