Update README.md

README.md

@@ -20,12 +20,12 @@ Given two sentences (a premise and a hypothesis), the model outputs the logits o
 
 You can test the model using the HuggingFace model widget on the side:
 - Input two sentences (premise and hypothesis) one after the other.
-- The model returns the probabilities of 3 labels: entailment, neutral and contradiction respectively.
+- The model returns the probabilities of 3 labels: entailment(LABEL:0), neutral(LABEL:1) and contradiction(LABEL:2) respectively.
 
-To use the model locally: 
+To use the model locally on your machine: 
 ```
-import torch
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+# import torch
+# device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 from transformers import AutoTokenizer, AutoModelForSequenceClassification