princhman commited on
Commit
8d284aa
·
verified ·
1 Parent(s): e1710da

fix minor typo

Browse files
Files changed (1) hide show
  1. mineru_single.py +2 -2
mineru_single.py CHANGED
@@ -191,6 +191,7 @@ def call_gemini_for_image_description(image_data: bytes) -> str:
191
  # Generate content with proper image format
192
  response = client.models.generate_content(
193
  model="gemini-2.0-flash",
 
194
  contents=[
195
  {
196
  "parts": [
@@ -204,8 +205,7 @@ If there is no text on this image, return the description of the image. 20 words
204
  "data": base64.b64encode(image_data).decode('utf-8')
205
  }
206
  }
207
- ],
208
- config=types.GenerateContentConfig(temperature=0.)
209
  }
210
  ]
211
  )
 
191
  # Generate content with proper image format
192
  response = client.models.generate_content(
193
  model="gemini-2.0-flash",
194
+ config=types.GenerateContentConfig(temperature=0.),
195
  contents=[
196
  {
197
  "parts": [
 
205
  "data": base64.b64encode(image_data).decode('utf-8')
206
  }
207
  }
208
+ ]
 
209
  }
210
  ]
211
  )