Spaces:
Runtime error
Runtime error
from transformers import pipeline | |
import gradio as gr | |
pretrained_name = "cahya/bert-base-indonesian-NER" | |
ner = pipeline( | |
"ner", | |
model=pretrained_name, | |
tokenizer=pretrained_name | |
) | |
examples = [ | |
"Ada apa dengan Jokowi di istana negara?", | |
] | |
def ner(text): | |
output = ner(text) | |
return {"text": text, "entities": output} | |
demo = gr.Interface(ner, | |
gr.Textbox(placeholder="Enter sentence here..."), | |
gr.HighlightedText(), | |
examples=examples) | |
if __name__ == "__main__": | |
demo.launch() |