Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
from typing import Callable, Dict, Literal, Union | |
import gradio as gr | |
def get_app( | |
models: list[str], | |
default_model: str, | |
src: Union[Callable[[str, str | None], gr.Blocks], Literal["models"], Dict[str, gr.Blocks]], | |
accept_token: bool = False, | |
dropdown_label: str = "Select Model", | |
**kwargs, | |
) -> gr.Blocks: | |
def update_model(new_model: str) -> list[gr.Column]: | |
return [gr.Column(visible=model_name == new_model) for model_name in models] | |
with gr.Blocks(fill_height=True) as demo: | |
model = gr.Dropdown(label=dropdown_label, choices=models, value=default_model) | |
columns = [] | |
for model_name in models: | |
with gr.Column(visible=model_name == default_model) as column: | |
if isinstance(src, dict): | |
src[model_name].render() | |
else: | |
gr.load(name=model_name, src=src, accept_token=accept_token, **kwargs) | |
columns.append(column) | |
model.change( | |
fn=update_model, | |
inputs=model, | |
outputs=columns, | |
api_name=False, | |
queue=False, | |
) | |
for fn in demo.fns.values(): | |
fn.api_name = False | |
return demo |