FilesExpand file tree

 main

/

gemini_pro_basic_example.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

43 lines (32 loc) · 1.31 KB

 main

/

gemini_pro_basic_example.py

Copy path

Top

File metadata and controls

• Code
• Blame

43 lines (32 loc) · 1.31 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

# Copyright 2023 Google LLC

#

# Licensed under the Apache License, Version 2.0 (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

#

# https://www.apache.org/licenses/LICENSE-2.0

#

# Unless required by applicable law or agreed to in writing, software

# distributed under the License is distributed on an "AS IS" BASIS,

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

# See the License for the specific language governing permissions and

# limitations under the License.

import os

PROJECT_ID = os.getenv("GOOGLE_CLOUD_PROJECT")

def generate_text() -> None:

# [START generativeaionvertexai_gemini_pro_example]

import vertexai

from vertexai.generative_models import GenerativeModel, Part

# TODO(developer): Update project_id and location

vertexai.init(project=PROJECT_ID, location="us-central1")

model = GenerativeModel("gemini-1.5-flash-002")

image_file = Part.from_uri(

"gs://cloud-samples-data/generative-ai/image/scones.jpg", "image/jpeg"

)

# Query the model

response = model.generate_content([image_file, "what is this image?"])

print(response.text)

# [END generativeaionvertexai_gemini_pro_example]

return response.text

if __name__ == "__main__":

generate_text()