Update README.md
Browse files
README.md
CHANGED
@@ -33,8 +33,8 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
|
33 |
import torch
|
34 |
|
35 |
hf_model_name= 'Muddassar/longformer-base-sentiment-5-classes'
|
36 |
-
model = AutoModelForSequenceClassification.from_pretrained(
|
37 |
-
tokenizer = AutoTokenizer.from_pretrained(
|
38 |
|
39 |
text = "Your text here"
|
40 |
inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=1500)
|
|
|
33 |
import torch
|
34 |
|
35 |
hf_model_name= 'Muddassar/longformer-base-sentiment-5-classes'
|
36 |
+
model = AutoModelForSequenceClassification.from_pretrained(hf_model_name)
|
37 |
+
tokenizer = AutoTokenizer.from_pretrained(hf_model_name)
|
38 |
|
39 |
text = "Your text here"
|
40 |
inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=1500)
|