kshitijthakkar commited on
Commit
b7d0706
·
1 Parent(s): b714b1e

fix: Update Gradio UI dropdown with all supported hardware options

Browse files

Updated 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.

Files changed (1) hide show
  1. app.py +14 -2
app.py CHANGED
@@ -281,9 +281,21 @@ def create_gradio_ui():
281
  )
282
  cost_hardware = gr.Dropdown(
283
  label="Hardware Type",
284
- choices=["auto", "cpu", "gpu_a10", "gpu_h200"],
 
 
 
 
 
 
 
 
 
 
 
 
285
  value="auto",
286
- info="'auto' will choose based on model type"
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