Update README.md

README.md

@@ -4,6 +4,8 @@ language:
 metrics:
 - f1
 pipeline_tag: text-classification
+datasets:
+- dariolopez/suicide-comments-es
 ---
 
 
@@ -14,15 +16,15 @@ This model is a fine-tuned version of [PlanTL-GOB-ES/roberta-base-bne](https://h
 # How to use
 
 ```python
-from transformers import pipeline
+>>> from transformers import pipeline
 
 
-model_name= 'dariolopez/roberta-base-bne-finetuned-suicide-es'
-pipe = pipeline("text-classification", model=model_name, device=0)
+>>> model_name= 'dariolopez/roberta-base-bne-finetuned-suicide-es'
+>>> pipe = pipeline("text-classification", model=model_name)
 
-pipe("Quiero acabar con todo. No merece la pena vivir.")
+>>> pipe("Quiero acabar con todo. No merece la pena vivir.")
 [{'label': 'Suicide', 'score': 0.9999703168869019}]
 
-pipe("El partido de fútbol fue igualado, disfrutamos mucho jugando juntos.")
+>>> pipe("El partido de fútbol fue igualado, disfrutamos mucho jugando juntos.")
 [{'label': 'Non-Suicide', 'score': 0.999990701675415}]
 ```