Arafath10 commited on
Commit
b71edf1
·
verified ·
1 Parent(s): daec3d5

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +4 -0
main.py CHANGED
@@ -16,6 +16,10 @@ from PIL import Image
16
  import io
17
  import requests
18
 
 
 
 
 
19
  secret = os.environ["gemini_key"]
20
  genai.configure(api_key=secret)
21
  model_vision = genai.GenerativeModel('gemini-pro-vision')
 
16
  import io
17
  import requests
18
 
19
+ 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')