Spaces:
Sleeping
Sleeping
from transformers import pipeline | |
import gradio as gr | |
get_completion = pipeline("ner", model="dslim/bert-base-NER") | |
def ner(input): | |
output = get_completion(input) | |
return {"text": input, "entities": output} | |
gr.close_all() | |
demo = gr.Interface(fn=ner, | |
inputs=[gr.Textbox(label="Text to find entities", lines=2)], | |
outputs=[gr.HighlightedText(label="Text with entities")], | |
title="NER with dslim/bert-base-NER", | |
description="Find entities using the `BERT-base` model under the hood!", | |
allow_flagging="never", | |
#Here we introduce a new tag, examples, easy to use examples for your application | |
examples=["My name is Raul and I live in Niterói, Rio de Janeiro, Brazil"]) | |
demo.launch() |