Update README.md

README.md

@@ -73,6 +73,10 @@ print(nlp(review_text))
 
 ```{python}
 
+import torch
+from transformers import AutoTokenizer, BertForSequenceClassification,TextClassificationPipeline
+from numpy import np
+
 model_name  = 'M47Labs/spanish_news_classification_headlines'
 MAX_LEN = 32
 
@@ -94,8 +98,8 @@ encoded_review = tokenizer.encode_plus(
   return_tensors='pt',
 )
 
-input_ids = encoded_review['input_ids']#.to(device)
-attention_mask = encoded_review['attention_mask']#.to(device)
+input_ids = encoded_review['input_ids']
+attention_mask = encoded_review['attention_mask']
 output = model(input_ids, attention_mask)
 
 _, prediction = torch.max(output['logits'], dim=1)