import torch
from transformers import AutoModel, AutoTokenizer

phobert = AutoModel.from_pretrained("hmthanh/VietnamLegalText-SBERT")
tokenizer = AutoTokenizer.from_pretrained("hmthanh/VietnamLegalText-SBERT")

sentence = 'Vượt đèn đỏ bị phạt bao nhiêu tiền?'  

input_ids = torch.tensor([tokenizer.encode(sentence)])

with torch.no_grad():
    features = phobert(input_ids)  # Models outputs are now tuples

hmthanh/VietnamLegalText-SBERT

Vietnamese Legal Text BERT

Table of contents

Using Vietnamese Legal Text BERT hmthanh/VietnamLegalText-SBERT

Using Vietnamese Legal Text BERT transformers

Installation

Pre-trained models

Example usage

hmthanh/
VietnamLegalText-SBERT

Using Vietnamese Legal Text BERT `hmthanh/VietnamLegalText-SBERT`

Using Vietnamese Legal Text BERT `transformers`