YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
StegoLSTM export
Exported from:
- checkpoint:
checkpoints/lm/stego-bilstm-word2vec-lm-lr0.0001-epoch=06-val_f1=0.82.ckpt - vocab:
checkpoints/lm/vocab.pkl
Usage
from 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
- Downloads last month
- 3
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support