adridonha commited on
Commit
59e75f2
·
1 Parent(s): 60b564d

actualizacion sentimientos

Browse files
Files changed (2) hide show
  1. app.py +18 -3
  2. requirements.txt +3 -0
app.py CHANGED
@@ -1,7 +1,22 @@
1
  import gradio as gr
 
2
 
3
- def greet(name):
4
- return "Hello " + name + "!!"
5
 
6
- demo = gr.Interface(fn=greet, inputs="text", outputs="text")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
7
  demo.launch()
 
1
  import gradio as gr
2
+ from transformers import pipeline
3
 
4
+ # Inicialización del pipeline con el modelo específico para análisis de sentimientos en español
5
+ clasificador = pipeline("sentiment-analysis", model="pysentimiento/robertuito-sentiment-analysis")
6
 
7
+ def puntuacion_sentimientos(texto):
8
+ resultado = clasificador(texto)
9
+ print(resultado) # Imprime el resultado crudo en la consola para depuración
10
+ etiqueta = resultado[0]["label"]
11
+
12
+ if(etiqueta == "POS"):
13
+ respuesta = "Tu frase es muy positiva"
14
+ elif etiqueta == "NEG":
15
+ respuesta = "Tu frase es muy negativa"
16
+ else:
17
+ respuesta = "ni fu ni fa" # Para casos neutros (NEU)
18
+
19
+ return respuesta
20
+
21
+ demo = gr.Interface(fn=puntuacion_sentimientos, inputs="text", outputs="text")
22
  demo.launch()
requirements.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ gradio==6.0.2
2
+ transformers==4.49.0
3
+ torch==2.6.0