supermy commited on
Commit
40b7064
1 Parent(s): 559a390

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -57,14 +57,14 @@ if __name__ == '__main__':
57
  sep_id = tokenizer.sep_token_id
58
  unk_id = tokenizer.unk_token_id
59
 
60
- model = GPT2LMHeadModel.from_pretrained("lewiswu1209/gpt2-chinese-composition")
61
  model.eval()
62
  gr.Interface(
63
  fn=generate,
64
  inputs=[
65
  "text",
66
- gr.Textbox(lines=7, placeholder="在这里输入一个开头。"),
67
  "number"
68
  ],
69
- outputs=gr.Textbox(lines=15, placeholder="这里会输出一段文字。")
70
  ).launch()
 
57
  sep_id = tokenizer.sep_token_id
58
  unk_id = tokenizer.unk_token_id
59
 
60
+ model = GPT2LMHeadModel.from_pretrained("supermy/poetry")
61
  model.eval()
62
  gr.Interface(
63
  fn=generate,
64
  inputs=[
65
  "text",
66
+ gr.Textbox(lines=7, placeholder="在这里输入一个开头:水陆草木之花,可爱者甚蕃。晋陶渊明独爱菊。" value="水陆草木之花,可爱者甚蕃。晋陶渊明独爱菊。",label="文本开始"),
67
  "number"
68
  ],
69
+ outputs=gr.Textbox(lines=15, placeholder="AI生成的文本显示在这里。",label="生成的文本")
70
  ).launch()