Views
No views yet
1import torch
2from transformers import AutoModel, AutoTokenizer
3
4vihealthbert = AutoModel.from_pretrained("BookingCare/bkcare-bert-pretrained")
5tokenizer = AutoTokenizer.from_pretrained("BookingCare/bkcare-bert-pretrained")
6
7line = "Bệnh viện chợ rẫy ở Thành phố Hồ Chí Minh"
8
9input_ids = torch.tensor([tokenizer.encode(line)])
10with torch.no_grad():
11 features = vihealthbert(input_ids) # Models outputs are now tuples