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)
|