Update README.md
Browse files
README.md
CHANGED
|
@@ -69,6 +69,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_dir)
|
|
| 69 |
model = AutoModelForSequenceClassification.from_pretrained(model_dir)
|
| 70 |
model.eval()
|
| 71 |
|
|
|
|
| 72 |
text = "Apple stock surges after strong earnings report."
|
| 73 |
inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True, max_length=128)
|
| 74 |
with torch.no_grad():
|
|
|
|
| 69 |
model = AutoModelForSequenceClassification.from_pretrained(model_dir)
|
| 70 |
model.eval()
|
| 71 |
|
| 72 |
+
# Example
|
| 73 |
text = "Apple stock surges after strong earnings report."
|
| 74 |
inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True, max_length=128)
|
| 75 |
with torch.no_grad():
|