from model import models
import gradio as gr

if __name__ == "__main__":
    exclude_keys, interfaces, titles = ["interface"], [], []
    for model, args in models.items():
        interface = args["interface"]
        excluded_args = {k: args[k] for k in set(list(args.keys())) - set(exclude_keys)}
        interfaces.append(interface(**excluded_args))
        titles.append(model)

    # interface = gr.TabbedInterface(interfaces, titles, theme="soft")
    with gr.Blocks(theme="soft") as demo:
        gr.Markdown("# NusaBERT Demo")
        gr.TabbedInterface(interfaces, titles, theme="soft")
    demo.launch()