File size: 491 Bytes
c18feb7
5a877d0
0d5403b
 
c18feb7
0d5403b
c18feb7
0d5403b
 
5a877d0
0d5403b
5a877d0
0d5403b
5a877d0
1
2
3
4
5
6
7
8
9
10
11
12
13
14
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)