最近話題のGeminiをAPIで利用できるとのことで試してみました。
できること
- テキスト生成
- 画像とテキストからテキスト生成
- チャットでの会話
チャット
私の名前を覚えてもらい、それを答えさせる実験をしてみました。ちゃんと記憶してくれてますね！
コードはこちら
import google.generativeai as genai
def main():
genai.configure(api_key='API_KEY')
model = genai.GenerativeModel('gemini-pro')
chat = model.start_chat(history=[])
while True:
prompt = input('あなた: \n')
if prompt == 'exit':
break
print("_"*80)
response = chat.send_message(prompt, stream=True)
print('Gemini: ')
for chunk in response:
print(chunk.text)
print("_"*80)
