from transformers import AutoTokenizer,AutoModel,BertTokenizer
from transformers.pipelines import pipeline
import gradio as gr
from huggingface_hub import login
import os
login(os.environ["HF_Token"])
model = AutoModel.from_pretrained(
    "minskiter/resume-token-classification",
    use_auth_token=True,
    trust_remote_code=True
)
tokenizer = AutoTokenizer.from_pretrained(
    "hfl/chinese-bert-wwm"
)
ner_predictor = pipeline(
    task="nerpipe",
    model=model, 
    config=model.config,
    tokenizer=tokenizer,
    device="cpu",
    trust_remote_code=True
)

def ner_predictor_gradio(input):
    return ner_predictor(input)

demo = gr.Interface(fn=ner_predictor_gradio, inputs="text", outputs="text")
demo.launch()