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)