Nechba commited on
Commit
6be9120
·
verified ·
1 Parent(s): 19fedfb

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -34,7 +34,7 @@ def pdf_to_images(pdf_bytes: bytes) -> list[Image.Image]:
34
 
35
  def analyze_single_document(images: list, prompt: str) -> dict:
36
  """Analyze a single document and return results"""
37
- model = genai.GenerativeModel('learnlm-1.5-pro-experimental')
38
  response = model.generate_content([prompt] + images)
39
  return response.text
40
  def analyze_pdf_directly(pdf_bytes: bytes, prompt: str, model_name: str = "gemini-1.5-pro"):
 
34
 
35
  def analyze_single_document(images: list, prompt: str) -> dict:
36
  """Analyze a single document and return results"""
37
+ model = genai.GenerativeModel('gemini-2.5-pro-exp-03-25')
38
  response = model.generate_content([prompt] + images)
39
  return response.text
40
  def analyze_pdf_directly(pdf_bytes: bytes, prompt: str, model_name: str = "gemini-1.5-pro"):