fix
Browse files
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]
|
26 |
else:
|
27 |
model_key = f"qwen:{model_name}"
|
28 |
-
|
29 |
-
|
30 |
-
|
31 |
-
|
|
|
|
|
|
|
32 |
else:
|
33 |
gr.load(name=model_name, src=src, accept_token=accept_token, **kwargs)
|
34 |
-
|
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,
|