import openai
# APIキーを取得
openai.api_key = "YOUR_API_KEY"
#モデルを指定
model_engine = "text-davinci-002"
#指示(Prompt)を設定
prompt = "日本の首都はどこですか?"
#推論を実行
completions = openai.Completion.create(
engine=model_engine,
prompt=prompt,
max_tokens=1024,
n=1,
stop=None,
temperature=0.5,
)
#推論結果を出力
message = completions.choices[0].text
print(message)
|