title-instruction / README.md
zeusfsx's picture
Update README.md
3f691df
|
raw
history blame
482 Bytes
metadata
license: cc-by-sa-4.0
language:
  - uk
metrics:
  - f1
pipeline_tag: text-classification

Usage

from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline

tokenizer = AutoTokenizer.from_pretrained("zeusfsx/title-instruction")
model = AutoModelForSequenceClassification.from_pretrained("zeusfsx/title-instruction")


classification = pipeline("text-classification", model=model, tokenizer=tokenizer, device="mps")  # for mac I used mps