ozoora/kzlbert-3poi

Use:

tokenizer = BertTokenizerFast.frompretrained('ozooora/kzlbert-3poi') model = AutoModelForSequenceClassification.frompretrained('ozooora/kzlbert-3poi', return_dict=True)

@torch.nograd() def predict(text): inputs = tokenizer(text, maxlength=419, padding=True, truncation=True, returntensors='pt') outputs = model(**inputs) predictedprobs = torch.nn.functional.softmax(outputs.logits, dim=1) predicted = torch.argmax(predictedprobs, dim=1).item() return predicted, predictedprobs[0].tolist()