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:
|