NER-Demo / app.py
AltayNeset's picture
Upload app.py
a0d72bc verified
raw
history blame contribute delete
972 Bytes
# -*- coding: utf-8 -*-
"""GradioIleNER.ipynb
Automatically generated by Colab.
Original file is located at
https://colab.research.google.com/drive/1NfmayavArZT9V1I1LgMeAkM7QLmKAhXe
"""
# !pip install -q transformers
from transformers import pipeline
ner_pipeline = pipeline("ner", model="Tirendaz/roberta-base-NER")
text = "I am Tim and I work at Google"
ner_pipeline(text)
text_tr = "Benim adım Neşet ve Trendyol'da çalışıyorum"
ner_pipeline(text_tr)
ner_pipeline(text_tr, aggregation_strategy="simple")
def ner(text):
output = ner_pipeline(text, aggregation_strategy="simple")
return {"text": text, "entities": output}
# !pip install -q gradio
import gradio as gr
examples = ["My name is Tim and I live in California", "Ich arbeite bei Google in Berlin", "Neşet, Ankara'lı mı?"]
demo = gr.Interface(
ner,
gr.Textbox(placeholder="Enter text here..."),
gr.HighlightedText(),
examples=examples,
)
demo.launch(share= True)