Update app.py

app.py

@@ -1,16 +1,62 @@
 import gradio as gr
+from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
 
-'''
-def under_construction():
-    return "classificator under construction"
+# Importa aquí tus clasificadores de texto en inglés y español
+# Puedes usar modelos de tu elección y cargarlos usando herramientas como TensorFlow, PyTorch, etc.
 
-gr.Interface(fn=under_construction, inputs=None, outputs="text").launch()
-'''
+# Definir las funciones que se utilizarán para clasificar el texto en inglés y español
+def classify_english_text(text):
+    # Coloca aquí la lógica para clasificar el texto en inglés
+    # Puedes usar el clasificador de texto en inglés que tengas cargado
+    pipe = pipeline("text-classification", model="I2C-UHU/NT-Grief_EN")
+    label = pipe(text)[0]['label']
 
+    if label == "LABEL_1":
+      label = label + " --> Message with non-traumatic grief content"
 
-txt = "TEXT CLASSIFICATOR UNDER CONSTRUCTION"
+    elif label == "LABEL_0":
+      label = label + " --> Message without non-traumatic grief content"
 
-with gr.Blocks() as demo:
-    gr.Textbox(txt)
-    
-demo.launch()
+    return label, pipe(text)[0]['score']
+
+def classify_spanish_text(text):
+    # Coloca aquí la lógica para clasificar el texto en español
+    # Puedes usar el clasificador de texto en español que tengas cargado
+    pipe = pipeline("text-classification", model="I2C-UHU/NT-Grief_SP")
+    label = pipe(text)[0]['label']
+
+    if label == "LABEL_1":
+      label = label + " --> Message with non-traumatic grief content"
+
+    elif label == "LABEL_0":
+      label = label + " --> Message without non-traumatic grief content"
+
+    return label, pipe(text)[0]['score']
+
+# Función para redirigir a la función de clasificación correspondiente según el idioma seleccionado
+def classify_text(language, text):
+    if language == "English":
+        return classify_english_text(text)
+    elif language == "Español":
+        return classify_spanish_text(text)
+    else:
+        return "Por favor, seleccione un idioma válido / Please select a valid language.",0
+
+examples = [
+    ["el jefe del gobierno local dijo que la muerte de thomas schäfer puede estar relacionada con preocupaciones debido a la crisis por la pandemia del coronavirus."],
+    ["I don't know who but the way the coronavirus pandemic coronavirus pandemic is causing some people to be scared to death some people are dying of fear, it is worth remembering that death is worth remembering that death comes at the right precise moment, not before, not after after..."],
+]
+
+# Definir la interfaz Gradio
+iface = gr.Interface(
+    fn=classify_text,  # Como vamos a usar dos funciones diferentes, dejamos este valor como None
+    inputs=[
+        gr.Radio(["Español", "English"], label="Elija el idioma / Choose the language"),
+        gr.Textbox(label="Texto / Text")
+    ],
+    outputs=[gr.Textbox(label = "Prediction"), gr.Number(label = "Probability")],  # Salida de texto con el resultado de la clasificación
+    title = "Non Traumatic Grief Detector",
+)
+
+# Ejecutar la interfaz Gradio
+iface.launch()