harshithan
/

fb-post-classifier-roberta_v1

Text Classification

customer-support

Model card Files Files and versions Community

harshithan commited on May 22

Commit

20aa03f

·

verified ·

1 Parent(s): b400693

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -67,8 +67,8 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 from torch.nn.functional import softmax
 import torch
-model = AutoModelForSequenceClassification.from_pretrained("harshithan/fb-post-classifier-roberta")
-tokenizer = AutoTokenizer.from_pretrained("harshithan/fb-post-classifier-roberta")
 inputs = tokenizer("I love the fast delivery!", return_tensors="pt")
 outputs = model(**inputs)

 from torch.nn.functional import softmax
 import torch
+model = AutoModelForSequenceClassification.from_pretrained("harshithan/fb-post-classifier-roberta_v1")
+tokenizer = AutoTokenizer.from_pretrained("harshithan/fb-post-classifier-roberta_v1")
 inputs = tokenizer("I love the fast delivery!", return_tensors="pt")
 outputs = model(**inputs)