orozcohsu commited on
Commit
1fc4981
·
verified ·
1 Parent(s): 84dffdd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -21,6 +21,8 @@ def app():
21
  #model="/Users/admin/mypy/hugging-models/gpt2"
22
  if choice == '中文':
23
  model = 'uer/gpt2-chinese-cluecorpussmall'
 
 
24
  generator = pipeline("text-generation", model)
25
 
26
  # 生成文本
 
21
  #model="/Users/admin/mypy/hugging-models/gpt2"
22
  if choice == '中文':
23
  model = 'uer/gpt2-chinese-cluecorpussmall'
24
+ else:
25
+ model = 'uer/gpt2-chinese-cluecorpussmall'
26
  generator = pipeline("text-generation", model)
27
 
28
  # 生成文本