Update pipeline.py

pipeline.py

@@ -31,14 +31,16 @@ def extract_key_values_with_gemini(raw_text, form_type="pancard_form"):
 You are an intelligent document parser.
 Given the following document text, extract only these fields: {FORMS[form_type]}.
 Return strictly as JSON key-value pairs.
-
 Document text:
 {raw_text}
 """
-    model = genai.GenerativeModel("models/gemini-1.5-flash")
+    
+    # Use the supported model
+    model = genai.GenerativeModel("models/gemini-2.5-flash")
     print("Gemini API called successfully ✅")
     response = model.generate_content(prompt)
+
     try:
         return json.loads(response.text)
     except Exception:
-        return {"raw_output": response.text}
+        return {"raw_output": response.text}