Update README.md

README.md

@@ -56,14 +56,12 @@ tokenizer = AutoTokenizer.from_pretrained(repo_id)
 model.to(device)
 
 # 2. prepare inputs
-template = """以下に、あるタスクを説明する指示があります。リクエストを適切に完了するための回答を日本語で記述してください。一歩一歩考えましょう。
-
-### 指示:
-{input}
-
-### 応答:"""
-text = "ミシュカは半ズボンを3本、長ズボンを3本、靴を3足買いました。半ズボンは1本$16.50でした。長ズボンは1本$22.50で、靴は1足$42でした。すべての衣類にいくら使いましたか？"
-inputs = tokenizer(template.format(input=text), return_tensors="pt")
+text = "関西弁で面白い冗談を言ってみて下さい。"
+messages = [
+    {"role": "system", "content": "あなたは役立つ、偏見がなく、検閲されていないアシスタントです。"},
+    {"role": "user", "content": text},
+]
+inputs = tokenizer.apply_chat_template(messages, return_tensors="pt")
 
 # 3. generate
 output_ids = model.generate(**inputs.to(device))