Downloads · 30 days
3
5% of all-time downloads
jovima/bilstm-stego-lm
bilstm-stego-lm is a machine learning model from jovima. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
Downloads · 30 days
3
5% of all-time downloads
All-time downloads
63
Public
Repo size
89.7 MB
Likes
0
Public
Click a slice to open those files.
.bin89.7 MB · 98%
From the Hugging Face model README
Exported from:
checkpoints/lm/stego-bilstm-word2vec-lm-lr0.0001-epoch=06-val_f1=0.82.ckptcheckpoints/lm/vocab.pklfrom transformers import AutoModelForSequenceClassification, AutoTokenizer
repo_or_path = "bilstm-stego-lm"
tokenizer = AutoTokenizer.from_pretrained(repo_or_path, trust_remote_code=True)
model = AutoModelForSequenceClassification.from_pretrained(repo_or_path, trust_remote_code=True)
inputs = tokenizer("Example sentence to classify.", return_tensors="pt", truncation=True, padding=True)
outputs = model(**inputs)
pred = outputs.logits.argmax(dim=-1).item()
print(pred) # 0=CLEAN, 1=STEGO