Nicolás Larenas commited on
Update ai_model.py
Browse files- ai_model.py +1 -1
ai_model.py
CHANGED
|
@@ -2,7 +2,7 @@ import google.generativeai as genai
|
|
| 2 |
import os
|
| 3 |
|
| 4 |
# Load Google API key from environment
|
| 5 |
-
GOOGLE_API_KEY = os.
|
| 6 |
genai.configure(api_key=GOOGLE_API_KEY)
|
| 7 |
model = genai.GenerativeModel('gemini-pro')
|
| 8 |
|
|
|
|
| 2 |
import os
|
| 3 |
|
| 4 |
# Load Google API key from environment
|
| 5 |
+
GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
|
| 6 |
genai.configure(api_key=GOOGLE_API_KEY)
|
| 7 |
model = genai.GenerativeModel('gemini-pro')
|
| 8 |
|