Views
No views yet
aubmindlab/bert-base-arabertv2 for binary end-of-utterance detection
in Saudi Arabic conversational speech.1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4tokenizer = AutoTokenizer.from_pretrained("YOUR_USERNAME/arabic-eou-bertv2")
5model = AutoModelForSequenceClassification.from_pretrained("YOUR_USERNAME/arabic-eou-bertv2")
6
7inputs = tokenizer("تمام خلاص", return_tensors="pt")
8prob = torch.softmax(model(**inputs).logits, dim=-1)[0,1].item()