Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
|
@@ -35,7 +35,7 @@ Return strictly as JSON key-value pairs.
|
|
| 35 |
Document text:
|
| 36 |
{raw_text}
|
| 37 |
"""
|
| 38 |
-
model = genai.GenerativeModel("gemini-1.5-flash")
|
| 39 |
print("Gemini API called successfully ✅")
|
| 40 |
response = model.generate_content(prompt)
|
| 41 |
try:
|
|
|
|
| 35 |
Document text:
|
| 36 |
{raw_text}
|
| 37 |
"""
|
| 38 |
+
model = genai.GenerativeModel("models/gemini-1.5-flash")
|
| 39 |
print("Gemini API called successfully ✅")
|
| 40 |
response = model.generate_content(prompt)
|
| 41 |
try:
|