Spaces:
Running on Zero
Running on Zero
Fix IndexError when model list is empty at startup
Browse filesGuard against empty get_available_models() result; print error for diagnosis.
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
- app.py +3 -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 |
-
|
| 146 |
-
|
|
|
|
| 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 |
|