Update README.md
Browse files
README.md
CHANGED
|
@@ -67,8 +67,8 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
|
| 67 |
from torch.nn.functional import softmax
|
| 68 |
import torch
|
| 69 |
|
| 70 |
-
model = AutoModelForSequenceClassification.from_pretrained("harshithan/fb-post-classifier-
|
| 71 |
-
tokenizer = AutoTokenizer.from_pretrained("harshithan/fb-post-classifier-
|
| 72 |
|
| 73 |
inputs = tokenizer("I love the fast delivery!", return_tensors="pt")
|
| 74 |
outputs = model(**inputs)
|
|
|
|
| 67 |
from torch.nn.functional import softmax
|
| 68 |
import torch
|
| 69 |
|
| 70 |
+
model = AutoModelForSequenceClassification.from_pretrained("harshithan/fb-post-classifier-roberta_v1")
|
| 71 |
+
tokenizer = AutoTokenizer.from_pretrained("harshithan/fb-post-classifier-roberta_v1")
|
| 72 |
|
| 73 |
inputs = tokenizer("I love the fast delivery!", return_tensors="pt")
|
| 74 |
outputs = model(**inputs)
|