mikemcrae25
/

newswire_classifiers

Text Classification

Model card Files Files and versions Community

mikemcrae25 commited on 23 days ago

Commit

eebd529

·

verified ·

1 Parent(s): 764e674

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -52,8 +52,8 @@ pip install transformers torch
 ```python
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
-model = AutoModelForSequenceClassification.from_pretrained("username/newswire_classifier/AP")
-tokenizer = AutoTokenizer.from_pretrained("username/newswire_classifier/AP")
 text = "(AP) President speaks at conference..."
 inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=128)

 ```python
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
+model = AutoModelForSequenceClassification.from_pretrained("mike-mcrae/newswire_classifier/AP")
+tokenizer = AutoTokenizer.from_pretrained("mike-mcrae/newswire_classifier/AP")
 text = "(AP) President speaks at conference..."
 inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=128)