Update api/views.py

api/views.py

@@ -146,7 +146,7 @@ def dataExtract(request, link):
             },
         ]
 
-        model = genai.GenerativeModel(model_name='gemini-1.5-pro-latest', tools=entityTool, safety_settings=safety_settings)
+        model = genai.GenerativeModel(model_name='gemini-2.0-flash-lite', tools=entityTool, safety_settings=safety_settings)
         chat = model.start_chat()
         response = chat.send_message('PDF Data : \n\n'+OCRString)
         fc = response.candidates[0].content.parts[0].function_call