Spaces:
Runtime error
Runtime error
dariolopez
commited on
Commit
·
c441f7b
1
Parent(s):
bb3533c
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ from transformers import pipeline
|
|
3 |
import gradio as gr
|
4 |
|
5 |
|
6 |
-
hf_writer = gr.HuggingFaceDatasetSaver(os.getenv('HF_API_TOKEN'), "suicide-comment-es-space-
|
7 |
pipe = pipeline("text-classification", model='hackathon-somos-nlp-2023/roberta-base-bne-finetuned-suicide-es')
|
8 |
|
9 |
|
@@ -16,16 +16,20 @@ iface = gr.Interface(
|
|
16 |
inputs="text",
|
17 |
outputs="text",
|
18 |
allow_flagging="manual",
|
19 |
-
flagging_options=['Incorrect
|
20 |
flagging_callback=hf_writer,
|
21 |
examples=[
|
22 |
"Quiero acabar con todo. No merece la pena vivir.",
|
23 |
-
"El partido de fútbol fue igualado, disfrutamos mucho jugando juntos."
|
24 |
],
|
25 |
article="""
|
26 |
## Motivación
|
27 |
|
28 |
-
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
|
|
|
|
|
|
|
|
|
29 |
|
30 |
## Dataset
|
31 |
|
|
|
3 |
import gradio as gr
|
4 |
|
5 |
|
6 |
+
hf_writer = gr.HuggingFaceDatasetSaver(os.getenv('HF_API_TOKEN'), "suicide-comment-es-space-human-feedback")
|
7 |
pipe = pipeline("text-classification", model='hackathon-somos-nlp-2023/roberta-base-bne-finetuned-suicide-es')
|
8 |
|
9 |
|
|
|
16 |
inputs="text",
|
17 |
outputs="text",
|
18 |
allow_flagging="manual",
|
19 |
+
flagging_options=['Correct Classification', 'Incorrect Classification'],
|
20 |
flagging_callback=hf_writer,
|
21 |
examples=[
|
22 |
"Quiero acabar con todo. No merece la pena vivir.",
|
23 |
+
"El partido de fútbol fue igualado, todos disfrutamos mucho jugando juntos."
|
24 |
],
|
25 |
article="""
|
26 |
## Motivación
|
27 |
|
28 |
+
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.
|
29 |
+
|
30 |
+
## Feedback
|
31 |
+
|
32 |
+
Ayúdanos a mejorar la aplicación marcando la respuesta como Correcta o Incorrecta según corresponda. Nosotros usaremos el feedback para ampliar el dataset que se usará para fine-tunear el modelo en versiones del futuro.
|
33 |
|
34 |
## Dataset
|
35 |
|