taoki's picture
Update README.md
2103d14 verified
metadata
language:
  - ja
license: other
tags:
  - text-generation-inference
  - transformers
  - trl
  - deepseek
datasets:
  - sakusakumura/databricks-dolly-15k-ja-scored
  - nu-dialogue/jmultiwoz
  - kunishou/amenokaku-code-instruct
  - HachiML/alpaca_jp_python
license_name: deepseek
base_model: deepseek-ai/deepseek-coder-6.7b-instruct

Uploaded model

  • Developed by: taoki
  • License: deepseek
  • Finetuned from model : deepseek-ai/deepseek-coder-6.7b-instruct

Usage

from transformers import AutoTokenizer, AutoModelForCausalLM
import torch

tokenizer = AutoTokenizer.from_pretrained(
  "taoki/deepseek-coder-6.7b-it-jmultiwoz-dolly-amenokaku-alpaca_jp_python"
)
model = AutoModelForCausalLM.from_pretrained(
  "taoki/deepseek-coder-6.7b-it-jmultiwoz-dolly-amenokaku-alpaca_jp_python"
)

if torch.cuda.is_available():
    model = model.to("cuda")

prompt="""あなたはプログラミングアシスタントです。
### Instruction:
OpenCVを用いて定点カメラから画像を保存するコードを示してください。
### Response:
"""

input_ids = tokenizer(prompt, return_tensors="pt").to(model.device)
outputs = model.generate(
    **input_ids,
    max_new_tokens=512,
    do_sample=True,
    top_p=0.9,
    temperature=0.2,
    repetition_penalty=1.1,
)
print(tokenizer.decode(outputs[0]))

Output

<|begin▁of▁sentence|>あなたはプログラミングアシスタントです。
### Instruction:
OpenCVを用いて定点カメラから画像を保存するコードを示してください。
### Response:
以下のPythonコードは、OpenCVを使用して定点カメラから画像を取得し、保存する方法を示します。

```python
import cv2

# カメラ設定
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|>