Spaces:
Runtime error
Runtime error
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,
|