Update app.py

app.py

@@ -12,21 +12,22 @@ tokenizer = BertTokenizer.from_pretrained(".")
 model = TFGPT2LMHeadModel.from_pretrained(".")
 text_generator = TextGenerationPipeline(model, tokenizer) 
 
+title = "写一首诗"
+article = "<div style='text-align: center;'>使用的模型:<a href='https://huggingface.co/uer/gpt2-chinese-poem' target='_blank'>https://huggingface.co/uer/gpt2-chinese-poem</a></div>"
+
+
 def poem(cls, max_length):
     txt = text_generator('[CLS]'+cls, max_length=int(max_length), do_sample=True)
     return txt[0]['generated_text'].replace('[CLS]','').replace('[SEP]','').replace(' ','').replace('。','。\n') 
 
 iface = gr.Interface(
-  gr.Markdown(
-    """
-    # 写一首诗
-    模型：https://huggingface.co/uer/gpt2-chinese-poem
-    """),
   fn=poem, 
   inputs=[
-    gr.Textbox("今晚提测又加班，BUG一堆改不完。", lines=1, label="请输入一句诗词:"),
+    gr.Textbox("今晚提测又加班，BUG一堆改不完。", lines=1, label="请输入第一句:"),
     gr.Number(66, label='字数:'),
   ],
-  outputs=gr.Textbox(lines=6))
+  outputs=gr.Textbox(lines=6),
+  title=title,
+  article=article)
   
 iface.launch()