Files
sci-gui-agent-benchmark/mm_agents/gemini_test.py
Hilbert-Johnson 7560f4dc46 update SoM_agent
2023-12-31 19:13:17 +08:00

19 lines
445 B
Python

import PIL.Image
import google.generativeai as genai
genai.configure(api_key="AIzaSyANsETKHVo-D8jZu1SnTSaQgLOJEDgnj9Q")
# for m in genai.list_models():
# if 'generateContent' in m.supported_generation_methods:
# print(m.name)
model = genai.GenerativeModel('gemini-pro-vision')
img = PIL.Image.open('image.jpg')
messages = [
{'role':'user',
'parts': ["Explain this image.", img]}
]
response = model.generate_content(messages)