Spaces:
Running
Running
Update main.py
Browse files
main.py
CHANGED
@@ -10,7 +10,7 @@ if not GOOGLE_API_KEY:
|
|
10 |
|
11 |
# Configure Gemini model
|
12 |
genai.configure(api_key=GOOGLE_API_KEY)
|
13 |
-
gemini_model = genai.GenerativeModel('gemini-flash-
|
14 |
|
15 |
# Directory for storing PDFs (optional if you want to load raw text data)
|
16 |
PDF_DIRECTORY = 'data'
|
|
|
10 |
|
11 |
# Configure Gemini model
|
12 |
genai.configure(api_key=GOOGLE_API_KEY)
|
13 |
+
gemini_model = genai.GenerativeModel(model_name='gemini-1.5-flash-latest')
|
14 |
|
15 |
# Directory for storing PDFs (optional if you want to load raw text data)
|
16 |
PDF_DIRECTORY = 'data'
|