NusaBERT / app.py
StevenLimcorn's picture
Reformat code to be generic, adding new models in model.py
47a6c20
raw
history blame
476 Bytes
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)
demo = gr.TabbedInterface(interfaces, titles, theme="soft")
demo.launch(debug=True)