akhaliq HF staff commited on
Commit
fb47976
·
1 Parent(s): 0825b59
Files changed (2) hide show
  1. app_qwen.py +1 -1
  2. utils.py +8 -1
app_qwen.py CHANGED
@@ -19,7 +19,7 @@ demo = get_app(
19
  "qwen:qvq-72b-preview",
20
  ],
21
  default_model="qwen:qvq-72b-preview",
22
- src=ai_gradio.registry.values(),
23
  accept_token=not os.getenv("DASHSCOPE_API_KEY"),
24
  )
25
 
 
19
  "qwen:qvq-72b-preview",
20
  ],
21
  default_model="qwen:qvq-72b-preview",
22
+ src=ai_gradio.registry,
23
  accept_token=not os.getenv("DASHSCOPE_API_KEY"),
24
  )
25
 
utils.py CHANGED
@@ -21,7 +21,14 @@ def get_app(
21
  for model_name in models:
22
  with gr.Column(visible=model_name == default_model) as column:
23
  if isinstance(src, dict):
24
- src[model_name].render()
 
 
 
 
 
 
 
25
  else:
26
  gr.load(name=model_name, src=src, accept_token=accept_token, **kwargs)
27
  columns.append(column)
 
21
  for model_name in models:
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)