Test_Magus

Running

SergeyO7 commited on 18 days ago

Commit

858d834

verified ·

1 Parent(s): eeda61c

Update agent.py

Files changed (1) hide show

agent.py CHANGED Viewed

@@ -49,7 +49,7 @@ def ImageAnalysisTool(image_path: str) -> str:
     image = Image.open(BytesIO(response.content)).convert("RGB")
     model = LiteLLMModel(
-        model_id="gemini/gemini-2.0-flash-lite",
         api_key= os.environ.get("GEMINI_KEY"),
         max_tokens=8192
     )

     image = Image.open(BytesIO(response.content)).convert("RGB")
     model = LiteLLMModel(
+        model_id="gemini/gemini-2.0-flash",
         api_key= os.environ.get("GEMINI_KEY"),
         max_tokens=8192
     )