Create README.md

README.md

@@ -0,0 +1,25 @@
+** This is work in progress **
+
+Spanish News Classification Headlines
+
+SNCH: is a model for sequence classification trained on private dataset. The the model will output the probability of a label('ciencia_tecnologia','clickbait','cultura','deportes','economia','educacion','medio_ambiente', 'opinion','politica',
+ 'sociedad'), the arquitecture of the NN is only in Pytorch.
+ 
+
+Example 
+ 
+The model can be accessed simply as 'M47Labs/spanish_news_classification_headlines' using the Transformers library. An example on how to download and use the models next:
+  
+  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))