fix
Browse files
utils.py
CHANGED
|
@@ -22,9 +22,13 @@ def get_app(
|
|
| 22 |
with gr.Column(visible=model_name == default_model) as column:
|
| 23 |
if isinstance(src, dict):
|
| 24 |
if ':' in model_name: # Handle provider:model format
|
| 25 |
-
|
| 26 |
-
|
|
|
|
|
|
|
|
|
|
| 27 |
else:
|
|
|
|
| 28 |
src[model_name].render()
|
| 29 |
|
| 30 |
else:
|
|
|
|
| 22 |
with gr.Column(visible=model_name == default_model) as column:
|
| 23 |
if isinstance(src, dict):
|
| 24 |
if ':' in model_name: # Handle provider:model format
|
| 25 |
+
provider, model = model_name.split(':')
|
| 26 |
+
if f"{provider}:{model}" in src:
|
| 27 |
+
src[f"{provider}:{model}"].render()
|
| 28 |
+
else:
|
| 29 |
+
raise ValueError(f"Model {model_name} not found in registry. Available models: {list(src.keys())}")
|
| 30 |
else:
|
| 31 |
+
# Default provider handling if needed
|
| 32 |
src[model_name].render()
|
| 33 |
|
| 34 |
else:
|