Update README.md

README.md

@@ -30,16 +30,16 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
 tokenizer = AutoTokenizer.from_pretrained(
-  "deepseek-ai/deepseek-coder-6.7b-instruct"
+  "taoki/deepseek-coder-6.7b-it-jmultiwoz-dolly-amenokaku-alpaca_jp_python"
 )
 model = AutoModelForCausalLM.from_pretrained(
-  "deepseek-ai/deepseek-coder-6.7b-instruct"
+  "taoki/deepseek-coder-6.7b-it-jmultiwoz-dolly-amenokaku-alpaca_jp_python"
 )
 
 if torch.cuda.is_available():
     model = model.to("cuda")
 
-prompt="""You are an AI programming assistant, utilizing the DeepSeek Coder model, developed by DeepSeek Company, and you only answer questions related to computer science. For politically sensitive questions, security and privacy issues, and other non-computer science questions, you will refuse to answer.
+prompt="""あなたはプログラミングアシスタントです。
 ### Instruction:
 OpenCVを用いて定点カメラから画像を保存するコードを示してください。
 ### Response:
@@ -48,7 +48,7 @@ OpenCVを用いて定点カメラから画像を保存するコードを示し
 input_ids = tokenizer(prompt, return_tensors="pt").to(model.device)
 outputs = model.generate(
     **input_ids,
-    max_new_tokens=256,
+    max_new_tokens=512,
     do_sample=True,
     top_p=0.9,
     temperature=0.2,
@@ -60,24 +60,39 @@ print(tokenizer.decode(outputs[0]))
 # Output
 
 ````
-<｜begin▁of▁sentence｜>You are an AI programming assistant, utilizing the DeepSeek Coder model, developed by DeepSeek Company, and you only answer questions related to computer science. For politically sensitive questions, security and privacy issues, and other non-computer science questions, you will refuse to answer.
+<｜begin▁of▁sentence｜>あなたはプログラミングアシスタントです。
 ### Instruction:
 OpenCVを用いて定点カメラから画像を保存するコードを示してください。
 ### Response:
+以下のPythonコードは、OpenCVを使用して定点カメラから画像を取得し、保存する方法を示します。
+
 ```python
 import cv2
-cap = cv2.VideoCapture(0) # カメラの設定
-fourcc = cv2.VideoWriter_fourcc(*'XVID') # 動画の形式
-out = cv2.VideoWriter('output.avi', fourcc, 20.0, (640, 480)) # 出力先、fps、解像度
-while True:
-    ret, frame = cap.read() # 映像読み込み
-    if not ret: break
-    out.write(frame) # 書き込み
-    cv2.imshow('Frame', frame) # 表示
-    if cv2.waitKey(1) & 0xFF == ord('q'): # qで終了
-        break
+
+# カメラ設定
+camera_id = 0 # 通常のカメラIDは0
+cap = cv2.VideoCapture(camera_id)
+
+if not cap.isOpened():
+    print("Failed to open camera")
+else:
+    ret, frame = cap.read()
+    if ret:
+        # 画像を保存する
+        filename = 'image.jpg'
+        cv2.imwrite(filename, frame)
+        print('Image saved as', filename)
+    else:
+        print('Failed to capture image')
+
 cap.release()
 cv2.destroyAllWindows()
 ```
+
+上記のコードでは、`cv2.VideoCapture()`関数でカメラを指定し、`cap.read()`で画像を読み込みます。成功した場合はTrueが返り、画像がframeとして返されます。
+
+次に、`cv2.imwrite()`関数で画像を保存します。第一引数には保存先のパスを指定し、第二引数には保存する画像を指定します。
+
+最後に、`cap.release()`でカメラを解放し、`cv2.destroyAllWindows()`で全てのウィンドウを破棄します。
 <|EOT|>
 ````