Update README.md

README.md

@@ -52,12 +52,12 @@ Before classifying the input sequence, the phrase needs to be processed with the
 from transformers import BertModel, BertTokenizer
 
 tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
-input = "Short, sleeveless dress in an airy cottonweave that is open at the back with a tie."
+input = "Short, sleeveless dress in an airy cotton weave that is open at the back with a tie."
 
 texts = self.tokenizer(batch, padding='max_length', max_length = 512, truncation=True,return_tensors="pt")
 input_ids = texts["input_ids"]
 attention_mask = texts["attention_mask"]
 
 output = model(input_ids, attention_mask)
-class = output.argmax(dim=1)
+class = output.argmax(dim=1) # should be 13 (Dress)
 ```