Update app.py
Browse filesfixed image caption prompt
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
|
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:
|