akhaliq HF staff commited on
Commit
bf1ff30
·
1 Parent(s): 2e1c19d
Files changed (1) hide show
  1. utils.py +6 -2
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
- src[f"qwen:{model_name}"].render()
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: