Spaces:
Running
Running
| import gradio as gr | |
| from transformers import pipeline, AutoTokenizer, AutoModelForTokenClassification | |
| tokenizer = AutoTokenizer.from_pretrained("funmidab/mbeukman-finetuned") | |
| model = AutoModelForTokenClassification. from_pretrained ("funmidab/mbeukman-finetuned") | |
| ner = pipeline ("ner", model=model, tokenizer=tokenizer, aggregation_strategy="max") | |
| gr.Interface.from_pipeline(ner).launch() |