akhaliq HF staff commited on
Commit
7f0560a
·
1 Parent(s): fb47976
Files changed (1) hide show
  1. utils.py +9 -6
utils.py CHANGED
@@ -22,16 +22,19 @@ def get_app(
22
  with gr.Column(visible=model_name == default_model) as column:
23
  if isinstance(src, dict):
24
  if ':' in model_name:
25
- src[model_name].render()
26
  else:
27
  model_key = f"qwen:{model_name}"
28
- if model_key in src:
29
- src[model_key].render()
30
- else:
31
- src[model_name].render()
 
 
 
32
  else:
33
  gr.load(name=model_name, src=src, accept_token=accept_token, **kwargs)
34
- columns.append(column)
35
 
36
  model.change(
37
  fn=update_model,
 
22
  with gr.Column(visible=model_name == default_model) as column:
23
  if isinstance(src, dict):
24
  if ':' in model_name:
25
+ block = src[model_name]
26
  else:
27
  model_key = f"qwen:{model_name}"
28
+ block = src.get(model_key, src.get(model_name))
29
+
30
+ if isinstance(block, gr.Blocks):
31
+ block.render()
32
+ else:
33
+ # Handle the case where block is a function
34
+ block().render()
35
  else:
36
  gr.load(name=model_name, src=src, accept_token=accept_token, **kwargs)
37
+ columns.append(column)
38
 
39
  model.change(
40
  fn=update_model,