from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline

MODEL_NAME = "Lightblue/DeepSeek-R1-Distill-Qwen-7B-Japanese"

tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
model = AutoModelForCausalLM.from_pretrained(MODEL_NAME)

generator = pipeline("text-generation", model=model, tokenizer=tokenizer)

prompt = "こんにちは、これはテストです。"
result = generator(prompt, max_length=100)[0]['generated_text']
print(result)