Update README.md

README.md

@@ -48,13 +48,11 @@ def add_prefix(text, list_label, label_num = 20, shuffle = False):
 
 text_new, list_label_new = add_prefix(text,list_label,shuffle=False)
 
-ids = tokenizer.encode(text_new)
-tokens = tokenizer.convert_ids_to_tokens(ids)
-encoding = tokenizer([text],truncation=True, padding='max_length',max_length=512)
-item = {key: torch.tensor(val) for key, val in encoding.items()}
-logits = model(**item).logits
-probs = torch.nn.functional.softmax(logits, dim = -1).tolist()
-predictions = torch.argmax(logits, dim=-1)
+encoding = tokenizer([text_new],truncation=True, padding='max_length',max_length=512, return_tensors='pt')
+with torch.no_grad():
+    logits = model(**item).logits
+    probs = torch.nn.functional.softmax(logits, dim = -1).tolist()
+    predictions = torch.argmax(logits, dim=-1)
 ```