update readme
Browse files
README.md
CHANGED
|
@@ -77,6 +77,7 @@ inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True)
|
|
| 77 |
outputs = model(**inputs)
|
| 78 |
predictions = torch.argmax(outputs.logits, dim=-1)
|
| 79 |
print(predictions) # 0=Negative, 1=Neutral, 2=Positive
|
|
|
|
| 80 |
|
| 81 |
### Training hyperparameters
|
| 82 |
|
|
|
|
| 77 |
outputs = model(**inputs)
|
| 78 |
predictions = torch.argmax(outputs.logits, dim=-1)
|
| 79 |
print(predictions) # 0=Negative, 1=Neutral, 2=Positive
|
| 80 |
+
```
|
| 81 |
|
| 82 |
### Training hyperparameters
|
| 83 |
|