Update app.py
Browse files
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("
|
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()
|