Views
No views yet
checkpoints/lm/stego-bilstm-word2vec-lm-lr0.0001-epoch=06-val_f1=0.82.ckptcheckpoints/lm/vocab.pkl1from transformers import AutoModelForSequenceClassification, AutoTokenizer
2
3repo_or_path = "bilstm-stego-lm"
4tokenizer = AutoTokenizer.from_pretrained(repo_or_path, trust_remote_code=True)
5model = AutoModelForSequenceClassification.from_pretrained(repo_or_path, trust_remote_code=True)
6
7inputs = tokenizer("Example sentence to classify.", return_tensors="pt", truncation=True, padding=True)
8outputs = model(**inputs)
9pred = outputs.logits.argmax(dim=-1).item()
10print(pred) # 0=CLEAN, 1=STEGO