Spaces:
Sleeping
Sleeping
Update agentic.py
Browse files- agentic.py +2 -2
agentic.py
CHANGED
|
@@ -62,7 +62,7 @@ def get_general_model():
|
|
| 62 |
|
| 63 |
if llm_provider == "mistral":
|
| 64 |
general_model = ChatMistralAI(
|
| 65 |
-
model="
|
| 66 |
temperature=0,
|
| 67 |
max_retries=2,
|
| 68 |
api_key=os.getenv("MISTRAL_API_KEY")
|
|
@@ -257,7 +257,7 @@ def download_input_file(task_id: str) -> str:
|
|
| 257 |
|
| 258 |
######## LLM associations ########
|
| 259 |
|
| 260 |
-
general_model =
|
| 261 |
vision_model = get_vision_model()
|
| 262 |
video_handler_model = get_video_handler_model()
|
| 263 |
audio_handler_model = get_audio_handler_model()
|
|
|
|
| 62 |
|
| 63 |
if llm_provider == "mistral":
|
| 64 |
general_model = ChatMistralAI(
|
| 65 |
+
model="mistral-large-2411",#"ministral-8b-latest",#"mistral-small-latest",#
|
| 66 |
temperature=0,
|
| 67 |
max_retries=2,
|
| 68 |
api_key=os.getenv("MISTRAL_API_KEY")
|
|
|
|
| 257 |
|
| 258 |
######## LLM associations ########
|
| 259 |
|
| 260 |
+
general_model = get_vision_model()
|
| 261 |
vision_model = get_vision_model()
|
| 262 |
video_handler_model = get_video_handler_model()
|
| 263 |
audio_handler_model = get_audio_handler_model()
|