tellurion Claude Sonnet 4.6 commited on
Commit
115b3c7
·
1 Parent(s): 63d1fdd

Fix IndexError when model list is empty at startup

Browse files

Guard against empty get_available_models() result; print error for diagnosis.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Files changed (2) hide show
  1. app.py +3 -2
  2. backend/appfunc.py +2 -1
app.py CHANGED
@@ -142,8 +142,9 @@ def init_interface(opt, *args, **kwargs) -> None:
142
  pad_scale = gr.Slider(label="Reference padding scale", minimum=1, maximum=2, value=1, step=0.05)
143
 
144
  with gr.Row():
145
- sd_model = gr.Dropdown(choices=get_available_models(), label="Models",
146
- value=get_available_models()[0])
 
147
  extractor_model = gr.Dropdown(choices=line_extractor_list,
148
  label="Line extractor", value=default_line_extractor)
149
  mask_model = gr.Dropdown(choices=mask_extractor_list, label="Reference mask extractor",
 
142
  pad_scale = gr.Slider(label="Reference padding scale", minimum=1, maximum=2, value=1, step=0.05)
143
 
144
  with gr.Row():
145
+ available_models = get_available_models()
146
+ sd_model = gr.Dropdown(choices=available_models, label="Models",
147
+ value=available_models[0] if available_models else None)
148
  extractor_model = gr.Dropdown(choices=line_extractor_list,
149
  label="Line extractor", value=default_line_extractor)
150
  mask_model = gr.Dropdown(choices=mask_extractor_list, label="Reference mask extractor",
backend/appfunc.py CHANGED
@@ -37,7 +37,8 @@ def get_available_models():
37
  try:
38
  files = list_repo_files(HF_REPO_ID)
39
  return [f for f in files if f.endswith(".safetensors")]
40
- except Exception:
 
41
  return []
42
 
43
 
 
37
  try:
38
  files = list_repo_files(HF_REPO_ID)
39
  return [f for f in files if f.endswith(".safetensors")]
40
+ except Exception as e:
41
+ print(f"Failed to list models from {HF_REPO_ID}: {e}")
42
  return []
43
 
44