Update README.md
Browse files
README.md
CHANGED
|
@@ -59,7 +59,7 @@ tokenizer = RobertaTokenizer.from_pretrained(model_name)
|
|
| 59 |
|
| 60 |
# Example usage
|
| 61 |
inputs = tokenizer("This product is great!", return_tensors="pt")
|
| 62 |
-
outputs = model(**inputs)
|
| 63 |
|
| 64 |
# Get sentiment
|
| 65 |
logits = outputs.logits
|
|
|
|
| 59 |
|
| 60 |
# Example usage
|
| 61 |
inputs = tokenizer("This product is great!", return_tensors="pt")
|
| 62 |
+
outputs = model(**inputs) # 1 for positive, 0 for negative
|
| 63 |
|
| 64 |
# Get sentiment
|
| 65 |
logits = outputs.logits
|