harald-n-space / app.py
Harald Nilsen
new pipeline
0d5403b
raw
history blame contribute delete
491 Bytes
import gradio as gr
from transformers import pipeline
import torch
import warnings
ner_pipe = pipeline("ner", model="NbAiLab/nb-bert-base-ner", aggregation_strategy="simple")
def ner(text):
return ner_pipe(text)
ner_result = ner("Nærmeste samtlige biler og vogntog som passerte Ørskogfjellet da kontrollen pågikk ble stoppet. Hele 50 personer fra de ulike etatene var involvert i kontrollen.")
demo = gr.Interface(fn=ner, inputs="textbox", outputs="textbox")
demo.launch(share=True)