| import gradio as gr |
| import spacy |
| from spacy import displacy |
| from spacy.tokens import Span |
| from random import randint |
| import spacy |
| import random |
| from fastcoref import spacy_component |
| nlp = spacy.blank("th") |
|
|
| nlp.add_pipe( |
| "fastcoref", |
| config={'model_architecture': 'FCoref','model_path': 'pythainlp/han-coref-v1.0'} |
| ) |
|
|
| default = "แกว่าเราเข้มแข็งมากพอที่จะเห็นแกไปน่ารักกับใครหรอ" |
|
|
| def corefer(text): |
| doc = nlp(text) |
| tags = doc._.coref_clusters |
| title=None |
| dic_ents = {"text":text,"ents":[],"title":title} |
| _tag=[str(i) for i in list(range(len(tags)))] |
| for i,tag in enumerate(tags): |
| for s,e in tag: |
| dic_ents["ents"].append({"start": s, "end": e, "label": _tag[i]}) |
| colors={i:"#"+''.join([random.choice('0123456789ABCDEF') for j in range(6)]) for i in _tag} |
| return displacy.render(dic_ents, manual=True, style="ent",options=colors, page=True) |
|
|
|
|
| iface = gr.Interface(fn=corefer, |
| inputs=gr.Textbox(label="Enter Text To Corefer with FastCoref", lines=2, value=default), |
| outputs="html").queue() |
| iface.launch() |
