Hugging Face's logo

Spaces:
Nechba
/
csv-generation-img
Sleeping

App Files Community
Nechba commited on
Commit
0b8df33
·
verified ·
1 Parent(s): ea39242

Update utils.py

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