| import os | |
| import json | |
| import google.generativeai as genai | |
| from PIL import Image | |
| #loading the API key | |
| GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY") | |
| genai.configure(api_key= GOOGLE_API_KEY) | |
| #function to load gemini pro model | |
| def load_gemini_pro(): | |
| gemini_pro_model = genai.GenerativeModel("gemini-1.5-flash") | |
| return gemini_pro_model | |
| # Function to load image vision model | |
| def gemini_pro_vision_responce(prompt, image): | |
| gemini_pro_vision_model = genai.GenerativeModel("gemini-1.5-pro") | |
| responce = gemini_pro_vision_model.generate_content([prompt, image]) | |
| result = responce.text | |
| return result | |