garvitcpp commited on
Commit
fbaab7a
·
verified ·
1 Parent(s): 521c6a2

Update app/services/extraction.py

Browse files
Files changed (1) hide show
  1. app/services/extraction.py +1 -1
app/services/extraction.py CHANGED
@@ -359,7 +359,7 @@ Output:
359
  prompt += message["content"]
360
 
361
  # Configure the Gemini model
362
- model = genai.GenerativeModel('gemini-1.5-flash')
363
 
364
  # Generate response
365
  response = model.generate_content(prompt,
 
359
  prompt += message["content"]
360
 
361
  # Configure the Gemini model
362
+ model = genai.GenerativeModel('gemini-2.5-flash')
363
 
364
  # Generate response
365
  response = model.generate_content(prompt,