Update README.md

README.md

@@ -44,10 +44,11 @@ prompt="""以下は、タスクを説明する指示です。要求を適切に
 OpenCVを用いて定点カメラから画像を保存するコードを示してください。
 ### 応答:
 """
-
-input_ids = tokenizer(prompt, return_tensors="pt").to(model.device)
-outputs = model.generate(
-    **input_ids,
+input_ids = tokenizer.encode(prompt,
+                             add_special_tokens=False,
+                             return_tensors="pt")
+tokens = model.generate(
+    input_ids.to(device=model.device),
     max_new_tokens=128,
     do_sample=True,
     temperature=0.1,
@@ -55,8 +56,25 @@ outputs = model.generate(
     repetition_penalty=1.0,
     top_k=0
 )
-print(tokenizer.decode(outputs[0]))
+out = tokenizer.decode(tokens[0], skip_special_tokens=False)
+print(out)
 ```
 
 # Output
 
+````
+以下は、タスクを説明する指示です。要求を適切に満たす応答を書きなさい。
+
+### 指示:
+OpenCVを用いて定点カメラから画像を保存するコードを示してください。
+### 応答:
+```python
+import cv2
+
+# 画像を読み込む
+img = cv2.imread('image.jpg')
+
+# 画像を保存する
+cv2.imwrite('output.jpg', img)
+```<EOD|LLM-jp>
+````