Spaces:

bstraehle
/

grady

Running

bstraehle commited on May 3

Commit

cc5b408

verified ·

1 Parent(s): cdc9e59

Update crew.py

Files changed (1) hide show

crew.py CHANGED Viewed

@@ -18,13 +18,13 @@ from util import get_final_answer
 MANAGER_MODEL      = "gpt-4.1"
 AGENT_MODEL        = "gpt-4.1"
-FINAL_ANSWER_MODEL = "gpt-4.1"
-IMAGE_MODEL        = "gemini-2.0-pro"
-AUDIO_MODEL        = "gemini-2.0-pro"
-VIDEO_MODEL        = "gemini-2.0-pro"
-YOUTUBE_MODEL      = "gemini-2.0-pro"
-DOCUMENT_MODEL     = "gemini-2.0-pro"
 # LLM evaluation

 MANAGER_MODEL      = "gpt-4.1"
 AGENT_MODEL        = "gpt-4.1"
+FINAL_ANSWER_MODEL = "gemini-2.0-flash"
+IMAGE_MODEL        = "gemini-2.0-flash-thinking-exp-01-21"
+AUDIO_MODEL        = "gemini-2.0-flash-thinking-exp-01-21"
+VIDEO_MODEL        = "gemini-2.0-flash-thinking-exp-01-21"
+YOUTUBE_MODEL      = "gemini-2.0-flash-thinking-exp-01-21"
+DOCUMENT_MODEL     = "gemini-2.0-flash-thinking-exp-01-21"
 # LLM evaluation