Gemini API

GOOGLE COLAB NOTESHEET

SCRIPT(COLAB ONLY)

import pathlib

import textwrap

import google.generativeai as genai

from IPython.display import display

from IPython.display import Markdown

def to_markdown(text):

  text = text.replace('•', '  *')

  return Markdown(textwrap.indent(text, '> ', predicate=lambda _: True))

# Used to securely store your API key

from google.colab import userdata

# Or use `os.getenv('GOOGLE_API_KEY')` to fetch an environment variable.

GOOGLE_API_KEY=userdata.get('GOOGLE_API_KEY')

genai.configure(api_key=GOOGLE_API_KEY)

for m in genai.list_models():

  if 'generateContent' in m.supported_generation_methods:

    print(m.name)

model = genai.GenerativeModel('gemini-pro')

위 실행 후 새 코드 만들어서 다음 스크립트 실행

%%time

response = model.generate_content(input())

to_markdown(response.text)

input에 질문 넣으면 끝