Spaces:
Sleeping
Sleeping
Commit
·
6fba44c
1
Parent(s):
5ebc627
Fix model names to valid GPT-4o-mini and Gemini versions for improved compatibility
Browse files- mai_dx/main.py +3 -3
mai_dx/main.py
CHANGED
@@ -255,7 +255,7 @@ class MaiDxOrchestrator:
|
|
255 |
|
256 |
def __init__(
|
257 |
self,
|
258 |
-
model_name: str = "gpt-
|
259 |
max_iterations: int = 10,
|
260 |
initial_budget: int = 10000,
|
261 |
mode: str = "no_budget", # "instant", "question_only", "budgeted", "no_budget", "ensemble"
|
@@ -2604,13 +2604,13 @@ def run_mai_dxo_demo(
|
|
2604 |
orchestrator = MaiDxOrchestrator.create_variant(
|
2605 |
variant,
|
2606 |
budget=3000,
|
2607 |
-
model_name="
|
2608 |
max_iterations=5,
|
2609 |
)
|
2610 |
else:
|
2611 |
orchestrator = MaiDxOrchestrator.create_variant(
|
2612 |
variant,
|
2613 |
-
model_name="
|
2614 |
max_iterations=5,
|
2615 |
)
|
2616 |
|
|
|
255 |
|
256 |
def __init__(
|
257 |
self,
|
258 |
+
model_name: str = "gpt-4o-mini", # Fixed: Use valid GPT-4 Turbo model name
|
259 |
max_iterations: int = 10,
|
260 |
initial_budget: int = 10000,
|
261 |
mode: str = "no_budget", # "instant", "question_only", "budgeted", "no_budget", "ensemble"
|
|
|
2604 |
orchestrator = MaiDxOrchestrator.create_variant(
|
2605 |
variant,
|
2606 |
budget=3000,
|
2607 |
+
model_name="gemini/gemini-2.5-flash", # Fixed: Use valid model name
|
2608 |
max_iterations=5,
|
2609 |
)
|
2610 |
else:
|
2611 |
orchestrator = MaiDxOrchestrator.create_variant(
|
2612 |
variant,
|
2613 |
+
model_name="gemini/gemini-2.5-flash", # Fixed: Use valid model name
|
2614 |
max_iterations=5,
|
2615 |
)
|
2616 |
|