akhaliq HF staff commited on
Commit
128fb04
1 Parent(s): 6dd5aa6
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -24,7 +24,7 @@ def get_app(
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:
 
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}"](name=model_name, **kwargs)
28
  else:
29
  raise ValueError(f"Model {model_name} not found in registry. Available models: {list(src.keys())}")
30
  else: