ner-project / app.py
funmidab's picture
Create app.py
da956bd verified
Raw
History Blame Contribute Delete
390 Bytes
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()