pzangara commited on
Commit
9693593
·
1 Parent(s): e9c762e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -6
app.py CHANGED
@@ -4,6 +4,7 @@ import torch
4
  from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification
5
  import transformers
6
  import gradio as gr
 
7
 
8
  model_name = "MoritzLaurer/mDeBERTa-v3-base-xnli-multilingual-nli-2mil7"
9
  tokenizer = AutoTokenizer.from_pretrained(model_name)
@@ -13,17 +14,19 @@ def clasificador(input1, input2):
13
  sequence_to_classify = input1
14
  input2_list = input2.split(",") # Parse the input string as a list
15
  candidate_labels= [item.strip() for item in input2_list]
16
- output = classifier(sequence_to_classify, candidate_labels, multi_label=False)
17
- return output
 
 
18
 
19
 
20
  iface = gr.Interface(
21
  fn=clasificador,
22
  inputs=[gr.inputs.Textbox(label="Ingrese texto"),gr.inputs.Textbox(label="Ingrese lista de labels")],
23
- outputs=gr.outputs.Textbox(label="Clasificación"),
24
- examples=[
25
- ['Antes de que avancemos con el tema, si yo tengo las cuentas bloqueadas por un fraude, yo lo explique por quinta vez, ya está todo resuelto, el tema del fraude online que se hizo y demás, sí por un préstamo sí, pero tengo las cuentas bloqueadas, no solamente no puedo pagar Galicia, sino no puedo pagar nada, soy vos, me haces un favor si podes de alguna manera acelerar que me desactiven la el bloqueo de la cuenta para yo poder pagar todo lo que debo pagar, me haces un favor, ya me llamaron, anoche volví a repetir lo mismo, que evidentemente no quedó claro. Sí, estoy queriendo, hoy nuevamente, comunicarme con la oficial de cuenta, María soledad, de la sucursal de Ramos Mejía, si damos para tener una respuesta con respecto al desbloqueo de mi cuenta. E claro, si voy a tener que hablar con la oficial señora, nosotros Por lo pronto todavía no lo vemos, pero llamamos justamente para. A ver lo que te pido es que por favor no me llamen mas, sí hasta que no. Cheque de que la cuenta esté desbloqueada porque no pagó nada, pero no, Galicia no pago nada, ni siquiera la empleada. ¿Me entiendes porque tengo bloqueada la cuenta?','a,b,c'],
26
- ]
27
  )
28
 
29
  iface.launch()
 
4
  from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification
5
  import transformers
6
  import gradio as gr
7
+ from pysentimiento import create_analyzer
8
 
9
  model_name = "MoritzLaurer/mDeBERTa-v3-base-xnli-multilingual-nli-2mil7"
10
  tokenizer = AutoTokenizer.from_pretrained(model_name)
 
14
  sequence_to_classify = input1
15
  input2_list = input2.split(",") # Parse the input string as a list
16
  candidate_labels= [item.strip() for item in input2_list]
17
+ output1 = classifier(sequence_to_classify, candidate_labels, multi_label=False)
18
+ analyzer = create_analyzer(task="sentiment", lang="es")
19
+ output2=analyzer.predict(input1)
20
+ return output1, output2
21
 
22
 
23
  iface = gr.Interface(
24
  fn=clasificador,
25
  inputs=[gr.inputs.Textbox(label="Ingrese texto"),gr.inputs.Textbox(label="Ingrese lista de labels")],
26
+ outputs=[gr.outputs.Textbox(label="Clasificación"),gr.outputs.Textbox(label="Sentimiento")],
27
+ # examples=[
28
+ # ['Antes de que avancemos con el tema, si yo tengo las cuentas bloqueadas por un fraude, yo lo explique por quinta vez, ya está todo resuelto, el tema del fraude online que se hizo y demás, sí por un préstamo sí, pero tengo las cuentas bloqueadas, no solamente no puedo pagar Galicia, sino no puedo pagar nada, soy vos, me haces un favor si podes de alguna manera acelerar que me desactiven la el bloqueo de la cuenta para yo poder pagar todo lo que debo pagar, me haces un favor, ya me llamaron, anoche volví a repetir lo mismo, que evidentemente no quedó claro. Sí, estoy queriendo, hoy nuevamente, comunicarme con la oficial de cuenta, María soledad, de la sucursal de Ramos Mejía, si damos para tener una respuesta con respecto al desbloqueo de mi cuenta. E claro, si voy a tener que hablar con la oficial señora, nosotros Por lo pronto todavía no lo vemos, pero llamamos justamente para. A ver lo que te pido es que por favor no me llamen mas, sí hasta que no. Cheque de que la cuenta esté desbloqueada porque no pagó nada, pero no, Galicia no pago nada, ni siquiera la empleada. ¿Me entiendes porque tengo bloqueada la cuenta?','a,b,c'],
29
+ # ]
30
  )
31
 
32
  iface.launch()