TaskOnClass / app.py
Leo Liu
Update app.py
e264535 verified
raw
history blame
356 Bytes
import streamlit as st
from transformers import pipeline
# Load pipelines
pipe = pipeline("summarization", model="facebook/bart-large-cnn")
# Application
st.title("Make a summarization!")
user_input = st.text_input("")
if user_input:
result = pipe(user_input)
summarization = result[0]['summary_text']
st.write(f"Sentiment: {summarization}")