ner_example / app.py
cwchang's picture
Update app.py
e14dba9
import gradio as gr
from transformers import pipeline
ner_pipeline = pipeline(task='ner', model='cwchang/ner_model', device=-1)
def ner(text):
output = ner_pipeline(text, aggregation_strategy="simple")
return {"text": text, "entities": output}
demo = gr.Interface(
fn=ner,
inputs=gr.Textbox(placeholder="Please enter a sentence here..."),
outputs=gr.HighlightedText(),
examples=[
["Emily Johnson will attend a United Nations meeting in New York on December 5, 2023."],
["Microsoft plans to launch its new tech product in Tokyo at the beginning of 2024."],
["Professor Michael Brown from the University of Toronto will present his latest research on climate change."],
["The famous Christ the Redeemer statue in Rio de Janeiro, Brazil, is scheduled for renovation in July 2023."],
["Maria Fernandez will open a new art gallery in Paris in June 2023."],
["The 2023 World Economic Forum is set to take place in Davos, Switzerland, expected to attract numerous international leaders."],
["Japanese scientist Ichiro Tanaka discovered a new renewable energy technology at the University of Tokyo."],
["The International Red Cross will initiate a humanitarian aid project in Nairobi, Kenya, in May 2023."],
["The Sydney Opera House in Australia will host a special concert during Christmas 2023."],
["The history museum in Berlin, Germany, will organize a World War II exhibition in January 2024."],
["千娜將於2023年8月參加在台北舉行的國際貿易會議。"],
["華為公司計劃於2024年在上海推出其最新的智慧型手機。"],
["褚志崧教授,來自清華大學,將在國際物理學會議上發表其關於量子計算的研究成果。"],
["成都的熊貓基地將於2023年春季對外開放新的參觀區。"],
["著名歌手王菲將於2023年10月在香港舉辦個人音樂會。"],
["2023年亞洲發展銀行年會計劃在新加坡召開,預計將有多位亞洲國家的領導人出席。"],
["日本動畫公司吉卜力將在2024年春季在臺北舉辦動畫展覽。"],
["聯合國教科文組織計劃於2023年在曼谷舉辦一次關於教育發展的國際研討會。"],
["雪梨歌劇院計劃於2023年聖誕節期間舉辦一場特別的交響音樂會。"],
["2024年,韓國首爾將舉辦世界環境保護論壇。"],]
)
demo.launch(debug=True)