Gemini-proを使って簡単なチャットボットを作ってみよう！

Mon, 22 Jan 2024 00:00:00 +0000

生成AIがどんどん進化していく中でGoogleが新たなモデルを出しました。それが"Gemini-pro"です。

ちなみにこちらのページの下の画像リンクからGeminiが搭載されたBardを使うことができます。

チャットボットを作るにはAPIが必要なのでまずはAPIキーを取得します。キーはこちらから取得できます。画像の"Get API key in Google AI Studio"のリンクをクリックします。

ページが移動したら"Get API key"タブから"Create API key in new project"をクリックすればAPIキーを取得することができます。これでチャットボットの準備ができました。

次にGoogle Colaboratoryを起動します。APIキーの設定とモデルの取得と機械学習webアプリケーションを設定するだけで使うことができます。今回は"Gradio"を使うことにします。他のアプリもありますが、いつか使ってみたいですね。

コードは以下のように設定します。“YOU’RE_APIKEY"に取得したAPIキーを入れてください。

!pip install gradio

import gradio as gr
import google.generativeai as genai

# APIキーの設定
GOOGLE_API_KEY = “YOU’RE_APIKEY”
genai.configure(api_key=GOOGLE_API_KEY)

# Generative AIモデルの初期化
model = genai.GenerativeModel(“gemini-pro”)

# チャットボットの動作を定義
def chatbot(prompt):
# Generative AIモデルを使用して推論を実行
response = model.generate_content(prompt)
return response.text

# Gradio UIの設定
iface = gr.Interface(
fn=chatbot,
inputs=gr.Textbox(type=“text”, placeholder=“Press Enter to Submit”),
outputs=“text”,
live=True
)

チャットボット on Xainome Blog