mirror of
https://github.com/gradio-app/gradio.git
synced 2025-03-13 11:57:29 +08:00
c9b8ad80fd
* ner accept hf format * formatting * added ner guide * guide fixes * added unittests for highlighttext * formatting
18 lines
420 B
Python
18 lines
420 B
Python
from transformers import pipeline
|
|
|
|
import gradio as gr
|
|
|
|
ner_pipeline = pipeline("ner")
|
|
|
|
examples = [
|
|
"Does Chicago have any stores and does Joe live here?",
|
|
]
|
|
|
|
def ner(text):
|
|
output = ner_pipeline(text)
|
|
return {"text": text, "entities": output}
|
|
|
|
gr.Interface(ner,
|
|
gr.Textbox(placeholder="Enter sentence here..."),
|
|
gr.HighlightedText(),
|
|
examples=examples).launch() |