AnkitAI commited on
Commit
37e7aeb
1 Parent(s): ead6dda

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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