Update app.py

app.py

@@ -12,7 +12,7 @@ def haiku_generate():
     )
     model.eval()
 
-    prompt ="あなたは日本語で五七五の俳句を生成するAIです。解説はせず、俳句だけ出力してください。"
+    prompt ="五七五の日本語俳句を1つ出力してください。俳句のみ。解説・説明は禁止。"
     inputs = tokenizer(prompt,return_tensors="pt")
     if not torch.cuda.is_available():
         inputs ={k:v.to("cpu") for k,v in inputs.items()}
@@ -20,6 +20,8 @@ def haiku_generate():
         outputs = model.generate(
             **inputs,
             max_new_tokens=25,
+            eos_token_id=tokenizer.eos_token_id,  # 終わりを明示！
+            pad_token_id=tokenizer.pad_token_id,
             do_sample=True,
             temperature=0.7,
             top_p=0.9