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
|