demo_piece / load.py
saketh092's picture
Upload 10 files
b1ce4f1 verified
raw
history blame contribute delete
No virus
350 Bytes
import pathlib
import textwrap
import google.generativeai as genai
def output(text):
response = model.generate_content("Summarize this and also don't make this too short "+text)
return response
GOOGLE_API_KEY="AIzaSyCgX2dUhfUNGZ1aSrjUIlXNfor1ylBdw7I"
genai.configure(api_key=GOOGLE_API_KEY)
model = genai.GenerativeModel('gemini-pro')