book-summary / app.py
bqpxben's picture
Update app.py
c02a435
raw
history blame
No virus
519 Bytes
import streamlit as st
from transformers import pipeline
generate_summary = pipeline("summarization", model="bqpxben/book-summary-tools")
def main():
st.title("Book Summarization App")
# Get user input
user_input = st.text_area("Enter the text you want to summarize:", "")
if st.button("Summarize"):
# Perform summarization
summary = generate_summary(user_input)
# Display result
st.write("Summary:")
st.write(summary)
if __name__ == "__main__":
main()