NusaBERT / app.py
StevenLimcorn's picture
Added NERP tagging and document retrieval wikipedia
2ef1af8
raw
history blame
632 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)
# interface = gr.TabbedInterface(interfaces, titles, theme="soft")
with gr.Blocks(theme="soft") as demo:
gr.Markdown("# LazarusNLP Indonesian NLP Demo")
gr.TabbedInterface(interfaces, titles, theme="soft")
demo.launch()