alanchen1115 commited on
Commit
6c78047
·
verified ·
1 Parent(s): 91b9cb6

Update Image_text_generation.py

Browse files
Files changed (1) hide show
  1. Image_text_generation.py +1 -1
Image_text_generation.py CHANGED
@@ -58,7 +58,7 @@ class Image_text_Generator:
58
  def generate_txt_with_gemini(self, prompt,doc_url):
59
  doc_data = httpx.get(doc_url).content
60
 
61
- response = client.models.generate_content(
62
  model="gemini-2.0-flash",
63
  contents=[
64
  types.Part.from_bytes(data=doc_data, mime_type='application/pdf',),prompt]
 
58
  def generate_txt_with_gemini(self, prompt,doc_url):
59
  doc_data = httpx.get(doc_url).content
60
 
61
+ response = genai_client.models.generate_content(
62
  model="gemini-2.0-flash",
63
  contents=[
64
  types.Part.from_bytes(data=doc_data, mime_type='application/pdf',),prompt]