import gradio as gr
import wbgtopic

clf = wbgtopic.WBGDocTopic()


def fn(inputs):
    return clf.suggest_topics(inputs)


demo = gr.Interface(fn=clf.suggest_topics, inputs="text", outputs=gr.JSON(label="Suggested topics"))
demo.launch()