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