Text Classification
Transformers
PyTorch
Spanish
roberta
Inference Endpoints
dariolopez commited on
Commit
29309ab
·
1 Parent(s): 0320beb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +27 -1
README.md CHANGED
@@ -1 +1,27 @@
1
- Hello
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - es
4
+ metrics:
5
+ - f1
6
+ ---
7
+
8
+
9
+ # Model Description
10
+
11
+ This model is a fine-tuned version of [PlanTL-GOB-ES/roberta-base-bne](https://huggingface.co/PlanTL-GOB-ES/roberta-base-bne) to detect suicidal ideation/behavior on public comments (forums, twitter, etc.) in Spanish language.
12
+
13
+ # How to use
14
+
15
+ ```python
16
+ from transformers import pipeline
17
+
18
+
19
+ model_name= 'dariolopez/roberta-base-bne-finetuned-suicide-es'
20
+ pipe = pipeline("text-classification", model=model_name, device=0)
21
+
22
+ pipe("Quiero acabar con todo. No merece la pena vivir.")
23
+ [{'label': 'Suicide', 'score': 0.9999703168869019}]
24
+
25
+ pipe("El partido de fútbol fue igualado, disfrutamos mucho jugando juntos.")
26
+ [{'label': 'Non-Suicide', 'score': 0.999990701675415}]
27
+ ```