Hugging Face's logo

Spaces:
Nechba
/
csv-generation-img
Sleeping

App Files Community
Nechba commited on
Commit
3318b75
·
verified ·
1 Parent(s): d26168c

Update utils.py

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