Update app.py

app.py

@@ -3,6 +3,22 @@ from transformers import pipeline
 import time
 
 #Nombre_modelo = 'pysentimiento/robertuito-sentiment-analysis'
+#######################creacion de funciones######################
+def cargar_modelo(model_name):
+    return pipeline("zero-shot-classification", model=model_name, device=0)
+
+def graficar(result):
+    # Crear un gráfico de pastel
+    fig, ax = plt.subplots()
+    ax.pie(result["scores"], labels=result["labels"], autopct="%1.1f%%", startangle=90)
+    ax.axis("equal")  # Hace que el gráfico sea un círculo perfecto
+    # Mostrar el gráfico en Streamlit
+    st.pyplot(fig)    
+
+#######################fin creacion de funciones######################
+
+
+
 
 st.title("Ejercicio interfaz transformers pipeline con modelos Zero-shot")
 
@@ -15,8 +31,6 @@ model_options = [
 selected_model = st.selectbox("Selecciona un modelo", model_options)
 #funcion eleccion del modelo
 @st.cache_resource
-def cargar_modelo(model_name):
-    return pipeline("zero-shot-classification", model=model_name, device=0)
 
 #asignacion del modelo a usar
 classifier = cargar_modelo(selected_model)
@@ -59,7 +73,8 @@ if st.button("Clasificar"):
                     </div>
                     """,
                     unsafe_allow_html=True,
-                )        
+                )
+        graficar(result)
     else:
         st.warning("Por favor, ingresa una oración y categorías válidas.")