|
import gradio as gr |
|
from kiwipiepy import Kiwi |
|
|
|
kiwi = Kiwi(model_type='sbg', typos='basic') |
|
|
|
def correct(input: str) -> str: |
|
input = kiwi.glue(input.split("\n")) |
|
sentence = kiwi.space(input) |
|
tokens = kiwi.tokenize(sentence) |
|
return kiwi.join(tokens) |
|
|
|
if __name__ == '__main__': |
|
with gr.Blocks(title='귀여운교정기') as demo: |
|
with gr.Row(): |
|
with gr.Column(): |
|
input_text = gr.Textbox() |
|
btn = gr.Button() |
|
output_text = gr.Textbox() |
|
|
|
btn.click(correct, inputs=[input_text], outputs=output_text) |
|
example = """배고픈 |
|
데밥 |
|
시간좀알려 줘""" |
|
examples = gr.Examples([example], inputs=input_text) |
|
|
|
demo.launch() |