dariolopez commited on
Commit
c441f7b
1 Parent(s): bb3533c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -4
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-flagged-outputs")
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 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, 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 compotamiento suicida.
 
 
 
 
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