Arafath10 commited on
Commit
4e1655f
·
verified ·
1 Parent(s): 89818f5

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -20,7 +20,7 @@ from dotenv import load_dotenv
20
  # Load the environment variables from the .env file
21
  load_dotenv()
22
 
23
- secret = os.environ["gemini_key"]
24
  genai.configure(api_key=secret)
25
  model_vision = genai.GenerativeModel('gemini-pro-vision')
26
  model_text = genai.GenerativeModel('gemini-pro')
@@ -51,7 +51,7 @@ def encode_image(image):
51
 
52
  def vision(image):
53
  # OpenAI API Key
54
- api_key = os.environ["open_ai_key"]
55
 
56
 
57
  # Getting the base64 string
 
20
  # Load the environment variables from the .env file
21
  load_dotenv()
22
 
23
+ secret = os.environ["GEMINI"]
24
  genai.configure(api_key=secret)
25
  model_vision = genai.GenerativeModel('gemini-pro-vision')
26
  model_text = genai.GenerativeModel('gemini-pro')
 
51
 
52
  def vision(image):
53
  # OpenAI API Key
54
+ api_key = os.environ["OPEN_AI"]
55
 
56
 
57
  # Getting the base64 string