masonbarnes
/

discord-message-classifier

Text Classification

Trained with AutoTrain

Inference Endpoints

Model card Files Files and versions Community

masonbarnes commited on Nov 18, 2022

Commit

d8b4013

•

1 Parent(s): 19ff375

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -43,9 +43,9 @@ Or Python API:
 ```
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
-model = AutoModelForSequenceClassification.from_pretrained("Johnsmith382234/autotrain-message-classifier-2136768960", use_auth_token=True)
-tokenizer = AutoTokenizer.from_pretrained("Johnsmith382234/autotrain-message-classifier-2136768960", use_auth_token=True)
 inputs = tokenizer("I love AutoTrain", return_tensors="pt")

 ```
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
+model = AutoModelForSequenceClassification.from_pretrained("masonbarnes/discord-message-classifier", use_auth_token=True)
+tokenizer = AutoTokenizer.from_pretrained("masonbarnes/discord-message-classifier", use_auth_token=True)
 inputs = tokenizer("I love AutoTrain", return_tensors="pt")