Gopikanth123 commited on
Commit
edbf719
·
verified ·
1 Parent(s): 819caf9

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
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-1.0')
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'