Views
No views yet
1from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline
2
3tokenizer = AutoTokenizer.from_pretrained("modernBERT-4-disfluency")
4model = AutoModelForTokenClassification.from_pretrained("modernBERT-4-disfluency")
5
6detector = pipeline("token-classification", model=model, tokenizer=tokenizer)
7results = detector("I um think that uh this works well")