dariolopez's picture
Update app.py
cd9c9f3
import os
from transformers import pipeline
import gradio as gr
hf_writer = gr.HuggingFaceDatasetSaver(os.getenv('HF_API_TOKEN'), "suicide-comment-es-space-human-feedback")
pipe = pipeline("text-classification", model='hackathon-somos-nlp-2023/roberta-base-bne-finetuned-suicide-es')
def detect_suicide_comment(input):
result = pipe(input)[0]
# return {result['label']: result['score']}
return result
iface = gr.Interface(
fn=detect_suicide_comment,
inputs="text",
outputs="text", # gr.Label(num_top_classes=1, type="confidences"),
allow_flagging="manual",
flagging_options=['Correct Classification', 'Incorrect Classification'],
flagging_callback=hf_writer,
examples=[
"Quiero acabar con todo. No merece la pena vivir.",
"El partido de fútbol fue igualado, todos disfrutamos mucho jugando juntos."
],
article="""
## Motivación
* Cada año se suicidan casi 1.000.000 de personas en el mundo y por cada suicidio, se cometen unos 20 intentos.
* Cada año se suicidan unas 4000 personas en España, siendo la primera causa de muerte externa. Esto son unas 11 muertes al día.
* Entre jóvenes de 15 a 29 años de edad, el suicidio es la 2ª causa principal de muerte a nivel mundial.
* Cada suicidio es una tragedia. Los efectos sobre las familias, amigos y comunidades son terribles y de amplio alcance. Por cada suicidio consumado, al menos 6 seres queridos se ven directamente afectados.
## Demo
Introduce cualquier frase en español, un modelo de inteligencia artificial tratará de predecir si la frase contiene algún patrón indicativo de ideación o comportamiento suicida.
## Feedback
¡Ayúdanos a mejorar la aplicación marcando la respuesta como Correcta o Incorrecta según corresponda!
Nosotros usaremos este feedback para ampliar el dataset que se usará para fine-tunear el modelo en versiones posteriores.
¡Gracias!
## Dataset
https://huggingface.co/datasets/hackathon-somos-nlp-2023/suicide-comments-es
## Model
https://huggingface.co/hackathon-somos-nlp-2023/roberta-base-bne-finetuned-suicide-es
## Team
* [dariolopez](https://huggingface.co/dariolopez)
* [diegogd](https://huggingface.co/diegogd)
""",
title="Detector de comentarios con ideación suicida"
)
iface.launch()