Update app.py

app.py

@@ -1,28 +1,18 @@
 import gradio as gr
-
-# 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.
+from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
 
 # 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
-    return 0.86
+    pipe = pipeline("text-classification", model="I2C-UHU/NT-Grief_EN")
+    return pipe(text)[0]['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
-    return 0.65
-
-# Definir la interfaz Gradio
-iface = gr.Interface(
-    fn=None,  # Como vamos a usar dos funciones diferentes, dejamos este valor como None
-    inputs=[
-        gr.inputs.Radio(["English", "Español"], label="Elija el idioma / Choose the language"),
-        gr.inputs.Textbox(label="Texto / Text")
-    ],
-    outputs="text"  # Salida de texto con el resultado de la clasificación
-)
+    pipe = pipeline("text-classification", model="I2C-UHU/NT-Grief_SP")
+    return pipe(text)[0]['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):
@@ -31,10 +21,17 @@ def classify_text(language, text):
     elif language == "Español":
         return classify_spanish_text(text)
     else:
-        return "Por favor, seleccione un idioma válido / Please select a valid language."
+        return "Por favor, seleccione un idioma válido / Please select a valid language.",0
 
-# Conecta la función 'classify_text' a la interfaz como función predictiva
-iface.func = classify_text
+# 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=["text", "number"]  # Salida de texto con el resultado de la clasificación
+)
 
 # Ejecutar la interfaz Gradio
 iface.launch()