import streamlit as st
from transformers import pipeline

pipe = pipeline("text2text-generation", model="bastiango97/resumen_articulos_CNN")

text = st.text_area('Enter the article and I will summirize it for you:')

if text:
    # Use the first element of the list and directly access the 'generated_text' key
    generated_text = pipe(text)[0].get('generated_text')

    # Display the generated text
    st.text(generated_text)