Update app.py

app.py

@@ -12,15 +12,16 @@ tokenizer = BertTokenizer.from_pretrained(".")
 model = TFGPT2LMHeadModel.from_pretrained(".")
 text_generator = TextGenerationPipeline(model, tokenizer) 
 
-def poem(cls,max_length):
+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(' ','') 
+    return txt[0]['generated_text'].replace('[CLS]','').replace('[SEP]','').replace(' ','').replace('。','。\n') 
 
 iface = gr.Interface(
   fn=poem, 
   inputs=[
     gr.Textbox(lines=1, default="春眠不觉晓，",label="请输入一句诗词:"),
-    gr.Number(50,label='字数:',default=50),
+    gr.Number(50, label='字数:',default=50),
   ],
-  outputs="text")
+  outputs=gr.Textbox(lines=4))
+  
 iface.launch()