Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
2
3tokenizer = AutoTokenizer.from_pretrained("zeusfsx/title-instruction")
4model = AutoModelForSequenceClassification.from_pretrained("zeusfsx/title-instruction")
5
6
7classification = pipeline("text-classification", model=model, tokenizer=tokenizer, device="mps") # for mac I used mps