civerson916 commited on
Commit
60167ef
·
verified ·
1 Parent(s): f03c280

Update app.py

Browse files

fixed image caption prompt

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -29,7 +29,7 @@ visit_webpage_tool = VisitWebpageTool()
29
  final_answer = FinalAnswerTool()
30
 
31
  model = LiteLLMModel(
32
- model_id="gemini/gemini-2.0-flash",
33
  temperature=0.2,
34
  api_key=os.environ.get("GEMINI_KEY")
35
  )
@@ -56,7 +56,7 @@ class ImageDescriptionTool(Tool):
56
  try:
57
  image_file = self.client.files.upload(file=f"{file_name}")
58
  image_description = self.client.models.generate_content(
59
- model="gemini-2.0-flash", contents=["Describe this audio clip", image_file]
60
  )
61
  return image_description.text
62
  except Exception as e:
 
29
  final_answer = FinalAnswerTool()
30
 
31
  model = LiteLLMModel(
32
+ model_id="gemini/gemini-2.0-flash-lite",
33
  temperature=0.2,
34
  api_key=os.environ.get("GEMINI_KEY")
35
  )
 
56
  try:
57
  image_file = self.client.files.upload(file=f"{file_name}")
58
  image_description = self.client.models.generate_content(
59
+ model="gemini-2.0-flash", contents=["Describe this image file", image_file]
60
  )
61
  return image_description.text
62
  except Exception as e: