Commit
·
b7d0706
1
Parent(s):
b714b1e
fix: Update Gradio UI dropdown with all supported hardware options
Browse filesUpdated the cost estimation tab's hardware dropdown to include all
supported hardware options (was causing 'not in list' errors):
**Modal hardware:**
- cpu, gpu_t4, gpu_l4, gpu_a10, gpu_l40s, gpu_a100, gpu_a100_80gb,
gpu_h100, gpu_h200, gpu_b200
**HuggingFace Jobs hardware:**
- cpu-basic, cpu-upgrade
- t4-small, t4-medium
- l4x1, l4x4
- a10g-small, a10g-large, a10g-largex2, a10g-largex4
- a100-large
- v5e-1x1, v5e-2x2, v5e-2x4
This matches the hardware options supported by the estimate_cost tool in mcp_tools.py.
app.py
CHANGED
|
@@ -281,9 +281,21 @@ def create_gradio_ui():
|
|
| 281 |
)
|
| 282 |
cost_hardware = gr.Dropdown(
|
| 283 |
label="Hardware Type",
|
| 284 |
-
choices=[
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 285 |
value="auto",
|
| 286 |
-
info="'auto'
|
| 287 |
)
|
| 288 |
cost_button = gr.Button("💰 Estimate", variant="primary")
|
| 289 |
|
|
|
|
| 281 |
)
|
| 282 |
cost_hardware = gr.Dropdown(
|
| 283 |
label="Hardware Type",
|
| 284 |
+
choices=[
|
| 285 |
+
"auto",
|
| 286 |
+
# Modal
|
| 287 |
+
"cpu", "gpu_t4", "gpu_l4", "gpu_a10", "gpu_l40s",
|
| 288 |
+
"gpu_a100", "gpu_a100_80gb", "gpu_h100", "gpu_h200", "gpu_b200",
|
| 289 |
+
# HuggingFace Jobs
|
| 290 |
+
"cpu-basic", "cpu-upgrade",
|
| 291 |
+
"t4-small", "t4-medium",
|
| 292 |
+
"l4x1", "l4x4",
|
| 293 |
+
"a10g-small", "a10g-large", "a10g-largex2", "a10g-largex4",
|
| 294 |
+
"a100-large",
|
| 295 |
+
"v5e-1x1", "v5e-2x2", "v5e-2x4"
|
| 296 |
+
],
|
| 297 |
value="auto",
|
| 298 |
+
info="Supports Modal and HuggingFace Jobs hardware. 'auto' selects cpu-basic (API) or a10g-small (local)."
|
| 299 |
)
|
| 300 |
cost_button = gr.Button("💰 Estimate", variant="primary")
|
| 301 |
|