Interfaz

app.py

@@ -0,0 +1,33 @@
+import gradio as gr
+from datasets import load_dataset
+from transformers import pipeline
+from presentation import main_title, examples
+
+model_name= 'hackathon-pln-es/electricidad-small-discriminator-finetuned-clasificacion-comentarios-suicidas'
+
+def clasificar_comentarios(comentario):
+    cls= pipeline("text-classification", model=model_name)
+    return cls(comentario)[0]['label']
+
+if __name__ == "__main__":
+    gr.Interface(
+        fn=clasificar_comentarios,
+        inputs=[
+            gr.inputs.Textbox(
+                lines=10,
+                label="Comentario a analizar:",
+                placeholder="Ingrese el comentario por favor...",
+                optional=False,
+            ),
+        ],
+        outputs=[
+            gr.outputs.HTML(
+                label="Resultado:"
+            )
+        ],
+        description=main_title,
+        examples=examples,
+        theme="dark-huggingface",
+        thumbnail="None",
+        css="https://cdn.jsdelivr.net/npm/bootstrap@3.3.7/dist/css/bootstrap.min.css",
+    ).launch()

presentation.py

@@ -0,0 +1,23 @@
+main_title = """
+<a href="https://unl.edu.ec/">
+    <img src="http://drive.google.com/uc?export=view&id=19VTXgT7jDdPvXtpY3HaxRj7NdG6w4RYY"  style="max-width: 100%; max-height: 10%; width: 1920px; height: 280px; object-fit: fill">
+</a>
+<h1 style="text-color:white">Clasificador de comentarios Suicidas</h1>
+<p>La siguiente aplicación ha sido diseñada con la intención de clasificar un comentario en "Suicida" y "No Suicida", con el fin de que para futuros trabajos e investigaciones se lo pueda implementar en una situación real para prevenir los casos de suicidio.</p>
+"""
+
+
+examples = [
+    [
+        "No entiendo por qué no soy lo suficientemente bueno. Sé que hay muchas personas en este Subreddit que están en situaciones increíblemente difíciles y siento que mi problema es tan pequeño en comparación. Sin embargo, es algo que me pesa y me siento tan bajo que quiero morir. Soy una mujer de 30 años y nunca he sido afortunada cuando se trata de citas o relaciones, sin embargo, conocí a alguien el año pasado con el que sentí que estaba conectado. Nos divertimos mucho juntos y me encantaba estar a su alrededor, pero sentí que no era lo suficientemente buena como para ser su novia. Me siento como una idiota al ponerme en esta situación, pero no veo ninguna esperanza."
+    ],
+    [
+        "Me corté por primera vez hoy. Solo para sentirme un poco mejor. No puedo perdonarme ahora. Estoy acabando con todo poco a poco. Estoy tan inestable en este momento, incluso en la medicación, la terapia y el ejercicio. Sé que puedo llegar a mis amigos para apoyar y no quiero cargarlos. Solo tengo miedo, y solo estoy solo en este momento. No quiero morir, pero la muerte se ve más y más favorable. Solo quiero que la culpa y el dolor innecesarias desaparezcan."
+    ],
+    [
+        "Siento que solo estoy viviendo debido a otras personas, no porque tenga una voluntad de vivir. La idea de matarme me da paz. Todo lo que quiero hacer es terminar mi vida. A cualquiera que lea esto, gracias."    
+    ],
+    [
+        "He estado enojado y triste sin ninguna razón válida, y solo quiero terminarlo. Sólo he estado enojado con todo en mi vida y es demasiado. Estoy harto de despertarse y estar enojado todo el día sin nada. Literalmente, nada en mi vida vale la pena enojarse, pero estoy irracionalmente enojado. Quiero dejarlo salir, pero no quiero lastimar a la gente en mi vida. Así que simplemente eliminaré lo que puedo controlar, a mí. Espero que el mundo sea mejor sin mí."
+    ],
+]

requirements.txt

@@ -0,0 +1,3 @@
+torch
+transformers
+gradio