mqha commited on
Commit
63a6bfa
·
1 Parent(s): 83b5787

纠正model路径加uer/

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
  # 生成文本