Jhoeel commited on
Commit
226cf35
1 Parent(s): afd2a27

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +23 -0
app.py ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Importa librerias
2
+
3
+ from transformers import pipeline
4
+ import gradio as gr
5
+
6
+ # Inicialización de Modelo de Clasificación de Sentimiento
7
+ modelo = 'pysentimiento/robertuito-sentiment-analysis'
8
+ pipeline_de_clasificador = pipeline("text-classification",model=modelo)
9
+
10
+ def clasificador(texto):
11
+ """
12
+ Función que clasifica un texto utilizando un pipeline de clasificación.
13
+
14
+ Parámetros:
15
+ texto (str): El texto a clasificar.
16
+
17
+ Devuelve:
18
+ str: La clase asignada al texto.
19
+ """
20
+ return str(pipeline_de_clasificador(texto)[0]['label'])
21
+
22
+ iface = gr.Interface(fn=clasificador, inputs="texto", outputs="texto")
23
+ iface.launch()