Jhoeel commited on
Commit
d30e255
1 Parent(s): 0b8e5bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -11
app.py CHANGED
@@ -1,5 +1,4 @@
1
  # Importa librerias
2
-
3
  from transformers import pipeline
4
  import gradio as gr
5
 
@@ -8,16 +7,16 @@ 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()
 
1
  # Importa librerias
 
2
  from transformers import pipeline
3
  import gradio as gr
4
 
 
7
  pipeline_de_clasificador = pipeline("text-classification",model=modelo)
8
 
9
  def clasificador(texto):
 
 
 
 
 
 
 
 
 
10
  return str(pipeline_de_clasificador(texto)[0]['label'])
11
 
12
+ with gr.Blocks() as demo:
13
+ with gr.Row():
14
+ with gr.Column():
15
+ input = gr.Textbox(label="Texto a clasificar")
16
+ btn_clasificar = gr.Button(value="Clasificar")
17
+ with gr.Column():
18
+ output = gr.Textbox(label="Sentimiento")
19
+
20
+ btn_clasificar.click(clasificador, inputs=input, outputs=output)
21
+
22
  iface.launch()