File size: 1,260 Bytes
41e9fc8 c473a2b 41e9fc8 c473a2b 14a7500 c473a2b bd544b7 c473a2b bd544b7 c473a2b bd544b7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
#Importamos las librerias necesarias, en caso que quieras hacerlo en tu PC debes instar las librerias detalladas en requirements.txt
from transformers import pipeline
import gradio as gr
from transformers.utils import logging
logging.set_verbosity_error()
pipe = pipeline(
"zero-shot-classification",
model="MoritzLaurer/xtremedistil-l6-h256-mnli-fever-anli-ling-binary"
)
# Etiquetas fijas (algunos rubros de consultas bancarias que se me ocurrieron)
etiquetas = [
"Pr茅stamos",
"Tarjetas",
"Transferencias",
"Inversiones",
"Seguros",
"Beneficios"
"Aplicaci贸n App"
]
def clasificar_consulta(texto):
resultado = pipe(texto, candidate_labels=etiquetas)
salida = "\n".join(
[f"{label}: {round(score * 100, 2)}%" for label, score in zip(resultado["labels"], resultado["scores"])]
)
return salida
# Interfaz con Gradio
iface = gr.Interface(
fn=clasificar_consulta,
inputs=gr.Textbox(label="Consulta del cliente"),
outputs=gr.Textbox(label="Clasificaci贸n de la consulta por rubro"),
title="Clasificaci贸n de consultas hecha por clientes a su entidad financiera",
description="Clasific谩 una consulta o reclamo y el modelo asignar谩 un rubro."
)
iface.launch(share=True)
|