Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import AutoTokenizer,AutoModelForCausalLM
 import torch
-model_id = "rinna/japanese-gpt2-medium"
+model_id = "rinna/japanese-gpt2-small"
 tokenizer = AutoTokenizer.from_pretrained(model_id,use_fast=False,trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(
     model_id,
@@ -27,7 +27,7 @@ def haiku_generate():
             temperature=1.3,
             top_p=0.9
         )
-    return tokenizer.decode(outputs[0],skip_special_tokens=True)
+    return tokenizer.decode(outputs[0],skip_special_tokens=True).split("\n")[-2]
 with gr.Blocks() as demo:
     output = gr.Textbox(label="俳句")
     btn = gr.Button("生成")