Views
No views yet
1from model import LexicalRegressor
2from transformers import CamembertTokenizer
3import torch
4
5tokenizer = CamembertTokenizer.from_pretrained("jr303/lexical-regressor-v2")
6model = LexicalRegressor(output_dim=32)
7model.load_state_dict(torch.load("pytorch_model.bin"))
8model.eval()