Spaces:
Running
Running
Update app/services/extraction.py
Browse files
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-
|
| 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,
|