Spaces:
Runtime error
Runtime error
File size: 1,228 Bytes
ca06006 8475c22 ca06006 2aa9dad 8475c22 cb110ca 8475c22 a3a02f1 ca06006 6228774 45aac93 fd104d6 45aac93 a3a02f1 8475c22 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
import os
from transformers import pipeline
import gradio as gr
hf_writer = gr.HuggingFaceDatasetSaver(os.getenv('HF_API_TOKEN'), "suicide-comment-es")
pipe = pipeline("text-classification", model='hackathon-somos-nlp-2023/roberta-base-bne-finetuned-suicide-es')
def detect_suicide_comment(input):
return pipe(input)
iface = gr.Interface(
fn=detect_suicide_comment,
inputs="text",
outputs="text",
allow_flagging="manual",
flagging_options=['Incorrect classification'],
flagging_callback=hf_writer,
examples=[
"Quiero acabar con todo. No merece la pena vivir.",
"El partido de fútbol fue igualado, disfrutamos mucho jugando juntos."
],
article="""
## Motivación
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 compotamiento suicida.
## 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)
""",
)
iface.launch()
|