victor HF staff commited on
Commit
d617adf
·
1 Parent(s): 2e463c2

refactor: Update model selection to use MODELS dictionary keys

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -301,11 +301,8 @@ with gr.Blocks() as demo:
301
  btn = gr.Button("Run")
302
  with gr.Accordion("Parameters", open=False):
303
  model_choice = gr.Radio(
304
- choices=[
305
- "Qwen/Qwen2.5-Coder-32B-Instruct",
306
- "deepseek-ai/DeepSeek-V3",
307
- ],
308
- value="deepseek-ai/DeepSeek-V3",
309
  label="Model",
310
  )
311
  top_p = gr.Slider(
 
301
  btn = gr.Button("Run")
302
  with gr.Accordion("Parameters", open=False):
303
  model_choice = gr.Radio(
304
+ choices=list(MODELS.keys()),
305
+ value=list(MODELS.keys())[0],
 
 
 
306
  label="Model",
307
  )
308
  top_p = gr.Slider(