Spaces:
Sleeping
Sleeping
Update retriever.py
Browse files- retriever.py +2 -1
retriever.py
CHANGED
|
@@ -1,9 +1,10 @@
|
|
| 1 |
from PyPDF2 import PdfReader
|
| 2 |
from google import genai
|
| 3 |
import time
|
|
|
|
| 4 |
|
| 5 |
# set up authentication
|
| 6 |
-
api_key =
|
| 7 |
client = genai.Client(api_key=api_key)
|
| 8 |
|
| 9 |
# read in documents
|
|
|
|
| 1 |
from PyPDF2 import PdfReader
|
| 2 |
from google import genai
|
| 3 |
import time
|
| 4 |
+
import os
|
| 5 |
|
| 6 |
# set up authentication
|
| 7 |
+
api_key = os.getenv("GEMINI_API_KEY")
|
| 8 |
client = genai.Client(api_key=api_key)
|
| 9 |
|
| 10 |
# read in documents
|