Update README.md

README.md

@@ -46,9 +46,69 @@ Columns: idTask,task content 1,idTag,tag.
 
 ## Example of Use
 
+### Pipeline
 
+```{python}
 
-An example on how to use the model in this page can be found in this colab notebook: https://colab.research.google.com/drive/1XsKea6oMyEckye2FePW_XN7Rf8v41Cw_?usp=sharing
+import torch
+from transformers import AutoTokenizer, BertForSequenceClassification,TextClassificationPipeline
+
+
+review_text = 'los vehiculos que esten esperando pasajaeros deberan estar apagados para reducir emisiones'
+path = "M47Labs/spanish_news_classification_headlines"
+tokenizer = AutoTokenizer.from_pretrained(path)
+model = BertForSequenceClassification.from_pretrained(path)
+
+
+nlp = TextClassificationPipeline(task = "text-classification",
+                model = model,
+                tokenizer = tokenizer)
+
+print(nlp(review_text))
+
+```
+```[{'label': 'medio_ambiente', 'score': 0.5648820996284485}]```
+
+### Pytorch
+
+```{python}
+
+model_name  = 'M47Labs/spanish_news_classification_headlines'
+MAX_LEN = 32
+
+
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+model = AutoModelForSequenceClassification.from_pretrained(model_name)
+
+texto = "las emisiones estan bajando, debido a las medidas ambientales tomadas por el gobierno"
+
+
+encoded_review = tokenizer.encode_plus(
+  texto,
+  max_length=MAX_LEN,
+  add_special_tokens=True,
+  #return_token_type_ids=False,
+  pad_to_max_length=True,
+  return_attention_mask=True,
+  return_tensors='pt',
+)
+
+input_ids = encoded_review['input_ids']#.to(device)
+attention_mask = encoded_review['attention_mask']#.to(device)
+output = model(input_ids, attention_mask)
+
+_, prediction = torch.max(output['logits'], dim=1)
+print(f'Review text: {texto}')
+
+print(f'Sentiment  : {model.config.id2label[prediction.detach().cpu().numpy()[0]]}')
+
+```
+```Review text: las emisiones estan bajando, debido a las medidas ambientales tomadas por el gobierno
+Sentiment  : medio_ambiente```
+
+
+A more in depth example on how to use the model can be found in this colab notebook: https://colab.research.google.com/drive/1XsKea6oMyEckye2FePW_XN7Rf8v41Cw_?usp=sharing
 
 
 ## Finetune Hyperparameters