mirror of
https://github.com/gradio-app/gradio.git
synced 2024-11-27 01:40:20 +08:00
db1ff439a7
* Fix docs * Fix ner demo * refactoring script templates * remove linksnav in guides Co-authored-by: aliabd <ali.si3luwa@gmail.com>
22 lines
464 B
Python
22 lines
464 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}
|
|
|
|
demo = gr.Interface(ner,
|
|
gr.Textbox(placeholder="Enter sentence here..."),
|
|
gr.HighlightedText(),
|
|
examples=examples)
|
|
|
|
if __name__ == "__main__":
|
|
demo.launch()
|