LymphSteamer commited on
Commit
a084d9e
·
verified ·
1 Parent(s): 8ecaac5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ def haiku_generate():
12
  )
13
  model.eval()
14
 
15
- prompt ="俳句だけを生成して"
16
  inputs = tokenizer(prompt,return_tensors="pt")
17
  if not torch.cuda.is_available():
18
  inputs ={k:v.to("cpu") for k,v in inputs.items()}
 
12
  )
13
  model.eval()
14
 
15
+ prompt ="あなたは日本語で五七五の俳句を生成するAIです。解説はせず、俳句だけ出力してください。"
16
  inputs = tokenizer(prompt,return_tensors="pt")
17
  if not torch.cuda.is_available():
18
  inputs ={k:v.to("cpu") for k,v in inputs.items()}