Update api.py

api.py

@@ -142,11 +142,8 @@ def extract_research_paper_arxiv(rp_number: str, node_type: str) -> dict:
             Here is the document: <document>{rp_data['abstract']}<document>"""
             
             try:
-                model_name = "gemini-2.5-flash-preview-05-20"
-                response = client.generate_content(
-                    model=model_name,
-                    contents=prompt
-                )
+                model = genai.GenerativeModel("gemini-2.5-flash-preview-05-20")
+                response = model.generate_content(prompt)
         
                 rp_data["summary"] = response.text
                 logger.info(f"Summary generated for Arxiv ID: {rp_number}")