Update README.md

README.md

@@ -24,7 +24,7 @@ encoded_input = tokenizer(text, return_tensors='pt')
 
 output = model(**encoded_input)
 
-scores = softmax(output.logits.cpu().detach().numpy())
+scores = softmax(output.logits.detach().numpy())
 
 id2class[np.argmax(scores)]