纠正model路径加uer/
Browse files
app.py
CHANGED
@@ -17,7 +17,8 @@ output_text = st.empty()
|
|
17 |
def generate_text(input_text):
|
18 |
# 加载预训练模型
|
19 |
if choice == '中文':
|
20 |
-
model = 'gpt2-chinese-cluecorpussmall' #
|
|
|
21 |
generator = pipeline("text-generation", model)
|
22 |
|
23 |
# 生成文本
|
|
|
17 |
def generate_text(input_text):
|
18 |
# 加载预训练模型
|
19 |
if choice == '中文':
|
20 |
+
model = 'uer/gpt2-chinese-cluecorpussmall' # 纠正后的应该可以
|
21 |
+
#model = 'gpt2-chinese-cluecorpussmall' # 会自动下载
|
22 |
generator = pipeline("text-generation", model)
|
23 |
|
24 |
# 生成文本
|